一次称重找出特殊球盒问题

有十个盒子，每个盒子10个球，有一个称重装置，如果已知球重量，假定分别是10g，11g，通过一次称重找出特殊的球盒

def find_special_box(boxes):
    # 从每个盒子中取出的球的数量
    balls_taken = [i + 1 for i in range(10)]

    # 实际的总重量
    actual_weight = 0
    for i, box in enumerate(boxes):
        for _ in range(i + 1):
            ball = box.pop()  # 假设 pop() 返回球的重量
            actual_weight += ball

    # 预期的总重量（如果所有球都是正常重量）
    # 我们需要先计算出正常球的重量
    normal_weight = None
    for i, box in enumerate(boxes):
        if len(box) > 0:
            normal_weight = box[0]
            break

    expected_weight = sum(balls_taken) * normal_weight

    # 计算差异
    special_box_index = abs(actual_weight - expected_weight)
    return special_box_index


# 示例数据
# 创建 10 个盒子，每个盒子有 10 个球
boxes = [[10] * 10 for _ in range(10)]
# 假设第 5 个盒子的球重量特殊
boxes[5] = [11] * 10

# 调用函数找出特殊盒子
special_box = find_special_box(boxes) - 1
print(f"Special box is: {boxes[special_box]}")
# 注: 只针对已知球重情况，若位置，对于像质量差异非质数的情况如4,6,8,9,无法定位到具体的盒子index