LCP 33. 蓄水

难度简单 19 收藏分享切换为英文接收动态反馈

给定 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:

    def storeWater(self, bucket: List[int], vat: List[int]) -> int:

        res_min = None

        for v in range(len(vat)):

            if vat[v] == 0 and v == len(vat)-1:

                return 0

            if vat[v] != 0:

                break

        add_index = None

        while 1:

            max_num = None

            if add_index == 0 or add_index:

                bucket[add_index] += 1

            for i in range(len(bucket)):

                a = bucket[i]

                if add_index == 0 or add_index:

                    c_num = 2

                else:

                    c_num = 1

                while a < vat[i]:

                    c_num+=1

                    if a==0:

                        a+=1

                    else:

                        a+=bucket[i]

                if not max_num:

                    max_num = c_num

                    add_index = i

                    continue

                if max_num and max_num 

                    max_num = c_num

                    add_index = i

            if not res_min:

                res_min = max_num

                continue

            if res_min and res_min > max_num:

                res_min = max_num

                continue

            if res_min and res_min <= max_num:

                break

        if not res_min:

            return 1

        return res_min