您当前的位置:首页 > 计算机 > 编程开发 > Python

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

时间:12-24来源:作者:点击数:

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

方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门