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

Python 解答:

import math
class Solution:
    def storeWater(self, bucket: List[int], vat: List[int]) -> int:
        min_total = 10000
        max_len = max(vat)
        if max_len == 0:
            return 0
        for i in range(1, max_len+1):
            total = i
            for j in range(len(vat)):
                num = math.ceil(vat[j]/i)
                if num > bucket[j]:
                    total += num-bucket[j]
            if total < min_total:
                min_total = total
        return min_total
最后修改日期: 2021年6月16日

留言

撰写回覆或留言

发布留言必须填写的电子邮件地址不会公开。