python编程题 你是一名医院的护士,粗心的你拿到药片后并没有进行分组,只记得要平均成两份,然而当你想起这件事的时候,病人的药片已经被分成了好几瓶。瓶字中的药片无法取出。鲜有n个药瓶,第二行n个数字代表每个药瓶中药片的数量,现在你想知道,是否可以将药片通过将药瓶组合的方式平均成两份(不允许拆分每瓶药片) 输入描述:首先输入一个数字n,代表药瓶的数量,第二行输入n个数字,代表每个瓶子中药片的数量nums
时间: 2023-05-28 22:01:40 浏览: 125
大一python编程题库和答案-Python编程练习题.pdf
思路:先计算出所有药片的总数,如果总数不能被2整除,则肯定不能平均成两份。否则,就计算出每份药片的数量,判断能否从所有瓶子中选出一些药瓶,使得这些药瓶中的药片数量之和等于每份药片的数量。
代码如下:
n = int(input())
nums = list(map(int, input().split()))
total = sum(nums) # 计算药片的总数
if total % 2 != 0: # 如果总数不能被2整除,则肯定不能平均成两份
print("NO")
else:
half = total // 2 # 计算每份药片的数量
dp = [0] * (half + 1) # 动态规划数组
dp[0] = 1 # base case
for num in nums:
for j in range(half, num - 1, -1):
dp[j] |= dp[j - num]
if dp[half]:
print("YES")
else:
print("NO")
阅读全文