问题描述:一位酒商共有5桶葡萄酒和1桶啤酒,6个桶的容量分别为30升、32升、36升、38升、40升和62升,并且只卖整桶酒,不零卖。第一位顾客买走了2整桶葡萄酒,第二位顾客买走的葡萄酒是第一位顾客的2倍。那么,本来有多少升啤酒呢?
解析:由于该酒商只卖整桶酒,简单分析几个桶的容量可知,第二位顾客必须买走剩下的3桶啤酒才有可能是第一位顾客的2倍。假设第一位顾客买走的葡萄酒共L升,那么第二位顾客买走的是2L升。也就是说,葡萄酒的总数应该能被3整除。
buckets = {30, 32, 36, 38, 40, 62}
for item in buckets:
if sum(buckets - {item}) % 3 == 0:
print(item)
break
上面是第一版本的代码,写完再看了一眼,觉得还是啰嗦了,不够Pythonic,于是改成了下面的代码。
buckets = {30, 32, 36, 38, 40, 62}
total = sum(buckets)
for item in buckets:
if (total-item) % 3 == 0:
print(item)
break
第二段代码比第一段代码效率略高,作为题目本身而言,做到这里就足够了,但是写代码的话还是要考虑的更全面一些。上面两段代码都没有考虑到一个问题,那就是酒商只卖整桶酒。如果考虑到这个问题的话,代码就变成了下面的样子。
buckets = {30, 32, 36, 38, 40, 62}
def solve(buckets):
#适用于6桶酒问题
assert len(buckets)==6
#酒的总量
total = sum(buckets)
#枚举法,逐个测试哪个是啤酒
for item in buckets:
div, mod = divmod((total-item), 3)
#葡萄酒数量应能被3整除
if mod == 0:
#除以3之后的整商应恰为某2桶酒之和
for i in buckets:
j = div-i
if j!=i and (j in buckets):
return (item, (i,j))
return 'no answer'
print(solve(buckets))
本题答案是40