[蓝桥杯] 分配口罩 python解法

试题 C: 分配口罩

【问题描述】

某市市长获得了若干批口罩，给定每批口罩的数量，市长要把口罩分配给市内的2所医院。

masks = [9090400, 8499400, 5926800, 8547000, 4958200, 4422600, 5751200, 4175600, 6309600, 5865200, 6604400, 4635000, 10663400, 8087200, 4554000]

由于物流限制，每一批口罩只能全部分配给其中一家医院。

市长希望2所医院获得的口罩总数之差越小越好。

请你计算这个差最小是多少？

结果：2400

import itertools

masks = [9090400, 8499400, 5926800, 8547000, 4958200, 4422600, 5751200, 4175600, 6309600, 5865200, 6604400, 4635000, 10663400, 8087200, 4554000]
ans = 99999999
s=sum(masks)
for i in itertools.combinations(masks,7):
    if ans > abs(s - sum(i) * 2):
        ans = abs(s-sum(i)*2)
print(ans)