Python求解啤酒问题（携程2016笔试题）

问题描述：一位酒商共有5桶葡萄酒和1桶啤酒，6个桶的容量分别为30升、32升、36升、38升、40升和62升，
并且只卖整桶酒，不零卖。第一位顾客买走了2整桶葡萄酒，第二位顾客买走的葡萄酒是第一位顾客的2倍。
那么，本来有多少升啤酒呢？
解析：由于该酒商只卖整桶酒，简单分析几个桶的容量可知，第二位顾客必须买走剩下的3桶葡萄酒才有可能是第一位顾客的2倍。
假设第一位顾客买走的葡萄酒共L升，那么第二位顾客买走的是2L升。也就是说，葡萄酒的总数应该能被3整除。所以，解法就呼之欲出了。

Python 解法1 

buckets = {30,32,36,38,40,62}
total = sum(buckets)
for item in buckets:
    if (total-item) % 3 == 0:
        print(item)
        break

虽然这样也能简单的解决问题，但是再考虑多一点，注意题目“只卖整桶”这个限制条件，更加规范的解法如下（但似乎没什么必要）

解法2

buckets = {30,32,36,38,40,62}
def solve(buckets):
    total = sum(buckets)
    for item in buckets:
        div,mod = divmod((total-item),3)
        if mod == 0:　　　　　　　　　　　　
            for i in buckets:
                j = div - i
                if j!=i and (j in buckets):
                    return (item,(i,j))
    return 'no answer'
    
print(solve(buckets))

代码中第8行 因为第一个顾客买的是2桶酒之和，所以验证是否存在这2桶酒，假如不存在即返回no answer。

Java 实现（java底子比较浅，写起来代码很难看，如果有更好的写法希望能赐教一下）

public static int bear(){
        int buckets[] = {30,32,36,38,40,62};
        int sum = 0;
        for(int bs :buckets){
            sum += bs;
        }
        for(int i=0;i<6;i++){
            int rest = sum-buckets[i];
            int mod = rest % 3;
            if(mod == 0) {
                int div = rest / 3;
                for(int j=0;j<6;j++){
                    int sep = div - buckets[j];
                    if(sep != buckets[j] && InBuckets(sep,buckets)){
                        return buckets[i];
                    }
                }
            }
        }
        return 0;
    }

public static boolean InBuckets(int x,int[] buckets){
        for(int i=0;i<buckets.length;i++){
            if(x == buckets[i]){
                return true;
            }
        }
        return false;
    }

参考资料 微信号 Python_xiaowu