0
点赞
收藏
分享

微信扫一扫

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

E_topia 2023-06-09 阅读 32


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

举报

相关推荐

0 条评论