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

输出样例:

60 1
70 2
80 3
90 4
30 1
40 5
10 1
20 2
5

解析:

简单来说,就是有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}")

提交结果:

猜你喜欢

转载自blog.csdn.net/weixin_51756038/article/details/134365023