leetcode-java-LCP 33. 蓄水

题目描述

给定 N 个无限容量且初始均空的水缸，每个水缸配有一个水桶用来打水，第 i 个水缸配备的水桶容量记作 bucket[i]。小扣有以下两种操作：

升级水桶：选择任意一个水桶，使其容量增加为 bucket[i]+1
蓄水：将全部水桶接满水，倒入各自对应的水缸
每个水缸对应最低蓄水量记作 vat[i]，返回小扣至少需要多少次操作可以完成所有水缸蓄水要求。

注意：实际蓄水量 达到或超过 最低蓄水量，即完成蓄水要求。

示例 1：
输入：bucket = [1,3], vat = [6,8]
输出：4
解释：
第 1 次操作升级 bucket[0]；
第 2 ~ 4 次操作均选择蓄水，即可完成蓄水要求。

示例 2：
输入：bucket = [9,0,1], vat = [0,2,2]
输出：3
解释：
第 1 次操作选择升级 bucket[1]
第 2~3 次操作选择蓄水，即可完成蓄水要求。

提示：
1 <= bucket.length == vat.length <= 100
0 <= bucket[i], vat[i] <= 10^4

看题解自己写的代码

思路：倒水是一起倒，升级是一个一个桶升级。所以倒水次数确定了就可以反推所有桶的升级次数之和
枚举倒水次数，则总次数=向上取整（缸容量/倒水次数）-桶容量+倒水次数

class Solution {
    public int storeWater(int[] bucket, int[] vat) {
        int max = 0;
        for(int v:vat)
            if(max<v) max=v;
        if(max==0) return 0;  //最大容量为0，代表不需蓄水，直接返回0
        int ans = Integer.MAX_VALUE;
        for(int pour_time=1;pour_time<max+bucket.length;pour_time++){
        if(pour_time>ans) break;   //减少不必要的循环
            int inc_time = 0,temp=0;
            for(int i=0;i<bucket.length;i++){   //枚举每个水桶，计算总升级次数
                if(vat[i]==0) continue;
                temp=(vat[i]+pour_time-1)/pour_time-bucket[i];  //容量/倒水次数-初始蓄水量=升级次数
                inc_time += Math.max(temp,0);;
                if(inc_time>ans) break;   //减少不必要的循环
            }
        ans = Math.min(pour_time+inc_time,ans);
        }
        return ans;
    }
}

剖析

1.题目不难，主要是思路想不到。没有想到先固定整体倒水次数，再根据倒水次数回推每个水桶需要提升容量的次数，再遍历一遍所有的桶。
2.细节，提交后会发现出问题。比如说：理论上应该对Math.ceil((double)vat[i]/pour_time)也就是每次需要的桶容量向上取整，向上取整又需要是浮点数，否则，整数除以整数永远是整数，整数不需要向上取整。但是又因为向上取整用时更长，所以换成加上（i-1）/i，一样可以达到效果。再比如：(vat[i]+pour_time-1)/pour_time-bucket[i]可能是一个负数，也就是只需要容量为2即可，实际上容量有9，那么这时应该返回0.又比如：要记得判断缸的容量是不是0，如果缸的容量是0那么不需要提升桶的容量,其实不判断也可以，答案不会错，但是内存消耗更大，因为要进行复杂的运算，判断容量是0然后直接跳出当前循环。还有增加一些判断来减少循环。