妙用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