python 字典类问题求解

Question

假定一款游戏中共有A,B,C,D四种商品，每种商品的金币价格存在字典price={'A':20,'B':30,'C':40,'D':50}中。编写一个程序，输入一个金币值，以及若干商品名称，判断若能购买这些商品，输出YES,否则输出NO。

测试1：（第1、2行为输入，第3行为输出）
40
A B
NO
测试2：（第1、2行为输入，第3行为输出）
100
C A B
YES

Answer 1

price = {'A':20, 'B':30, 'C':40, 'D':50}
money = input("Enter total money: ")
wish = input("Enter products delimited by space: ")
products = wish.split(' ')
bill = [price[p] for p in products]
total = sum(bill)
affordable = int(money) > total
print('Yes' if affordable else 'No')