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