Leetcode每日一题-2.12

难度:中等;解法:二分

1760. 袋子里最少数目的球

给你一个整数数组 nums ,其中 nums[i] 表示第 i 个袋子里球的数目。同时给你一个整数 maxOperations 。

你可以进行如下操作至多 maxOperations 次:

  • 选择任意一个袋子,并将袋子里的球分到 2 个新的袋子中,每个袋子里都有 正整数 个球。

    • 比方说,一个袋子里有 5 个球,你可以把它们分到两个新袋子里,分别有 1 个和 4 个球,或者分别有 2 个和 3 个球。

你的开销是单个袋子里球数目的 最大值 ,你想要 最小化 开销。

请你返回进行上述操作后的最小开销。

 

示例 1:

输入:nums = [9], maxOperations = 2
输出:3
解释:
- 将装有 9 个球的袋子分成装有 6 个和 3 个球的袋子。[9] -> [6,3] 。
- 将装有 6 个球的袋子分成装有 3 个和 3 个球的袋子。[6,3] -> [3,3,3] 。
装有最多球的袋子里装有 3 个球,所以开销为 3 并返回 3 。

示例 2:

输入:nums = [2,4,8,2], maxOperations = 4
输出:2
解释:
- 将装有 8 个球的袋子分成装有 4 个和 4 个球的袋子。[2,4,8,2] -> [2,4,4,4,2] 。
- 将装有 4 个球的袋子分成装有 2 个和 2 个球的袋子。[2,4,4,4,2] -> [2,2,2,4,4,2] 。
- 将装有 4 个球的袋子分成装有 2 个和 2 个球的袋子。[2,2,2,4,4,2] -> [2,2,2,2,2,4,2] 。
- 将装有 4 个球的袋子分成装有 2 个和 2 个球的袋子。[2,2,2,2,2,4,2] -> [2,2,2,2,2,2,2,2] 。
装有最多球的袋子里装有 2 个球,所以开销为 2 并返回 2 。

示例 3:

输入:nums = [7,17], maxOperations = 2
输出:7

思路

之前做过,但是后面又忘记了。看了一下之前做的属于二分法的做法,取一个范围表示最小开销可以达到的范围,最小值是1,最大值则为初始袋子里球的数目最大值。然后使用二分法获取mid,mid表示最小开销为mid,即初始每个袋子都需要分成最大数为mid的情况,而每个袋子需要的操作数就是Math.ceil(num/mid) ,对操作数进行累加,与maxOperations 进行对比,如果小于等于,则表示可以继续减小最小开销,否则需要增大最小开销。

代码

class Solution {
    public int minimumSize(int[] nums, int maxOperations) {
        int left=0, right=0;
        for (int num:nums) {
            right = Math.max(num, right);
        }
        int ans = right;
        while (left <= right) {
            int mid = (right-left)/2 + left;
            if (isValid(mid, nums, maxOperations)) {
                ans = Math.min(mid, ans);
                right=mid-1;
            } else {
                // right = mid-1;
                left = mid +1 ;
            }
        }

        return ans;
    }

    public boolean isValid(int target, int[] nums, int maxOperations) {
        int ops = 0;
        for (int num : nums) {
            if (num > target) {
                ops += Math.ceil(num * 1.0 / target)-1;
            }
        }
        // System.out.println(target +  " : " + ops);
        return ops <= maxOperations;
    }
}

LICENSED UNDER CC BY-NC-SA 4.0
Comment