妙用Python集合求解啤酒问题（携程2016笔试题）

问题描述：一位酒商共有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