pta 装箱问题 Python3

假设有N项物品，大小分别为s1、s2、…、si、…、sN，其中si为满足1≤si≤100的整数。要把这些物品装入到容量为100的一批箱子（序号1-N）中。装箱方法是：对每项物品, 顺序扫描箱子，把该物品放入足以能够容下它的第一个箱子中。请写一个程序模拟这种装箱过程，并输出每个物品所在的箱子序号，以及放置全部物品所需的箱子数目。

输入格式：

输入第一行给出物品个数N（≤1000）；第二行给出N个正整数si（1≤si≤100，表示第i项物品的大小）。

输出格式：

按照输入顺序输出每个物品的大小及其所在的箱子序号，每个物品占1行，最后一行输出所需的箱子数目。

输入样例：

8
60 70 80 90 30 40 10 20

输出样例：

解析:

简单来说，就是有N个容量为100的箱子，然后往里面尽可能的放物品——物品容量加起来不超过100就可以放一起。从输出来看它是从第一项开始装箱的，然后再判断后续物品能否放同一个箱子，如若不行则新增一个箱子。

Python代码：

# 获取物品个数
N = int(input())

# 获取每个物品的大小
sizes = list(map(int, input().split()))

# 初始化箱子列表，每个箱子的剩余容量为100
boxes = [100] * N

# 初始化记录箱子序号的列表
box_numbers = [-1] * N

# 记录所需的箱子数目
required_boxes = 0

for i in range(N):
    size = sizes[i]

    # 遍历箱子找到第一个能够容纳该物品的箱子
    for j in range(N):
        if boxes[j] >= size:
            box_numbers[i] = j + 1
            boxes[j] -= size
            break

    # 如果没有找到合适的箱子，需要新增一个箱子
    if box_numbers[i] == -1:
        required_boxes += 1
        box_numbers[i] = required_boxes
        boxes[required_boxes - 1] -= size

# 输出每个物品所在的箱子序号
for i in range(N):
    print(f"{sizes[i]} {box_numbers[i]}")

# 输出所需的箱子数目
required_boxes = max(box_numbers)
print(f"{required_boxes}")