【二分查找】leetcode 1760. 袋子里最少数目的球

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<=nums.length<=10^5$
• $1<=maxOperations,nums[i]<=10^9$

方法：二分查找

解题思路

由题意得，当单个袋子里球的数目越大时，则分球的次数就越少；当单个袋子里球的数目越少时，则分球的次数就越多。因此，本题具有单调性，可以利用二分查找来解题。

二分答案的左边界为 1，右边界为数组中最大值，如何判断 mid 是否符合条件？

• 如果当前 mid 使得分球次数小于等于最大分球次数，则区间缩小为 $[l,mid]$；
• 如果当前 mid 使得分球次数大于最大分球次数，则区间缩小为 $[mid+1,r]$。

直到找到使得分球次数小于等于最大分球次数的最小的单个袋子里球的数目。

代码

class Solution {
public:
    int check(vector<int>& nums, int mid, int maxOperations) {
        int operations = 0;
        for(int i = 0; i < nums.size(); i++) 
            operations += (nums[i] - 1) / mid;
        return operations <= maxOperations;
    }
    int minimumSize(vector<int>& nums, int maxOperations) {
        int l = 1, r = *max_element(nums.begin(), nums.end()), mid;
        while(l < r)
        {
            mid = (l + r) >> 1;
            if(check(nums, mid, maxOperations))    r = mid;
            else    l = mid + 1;
        }
        return l;
    }
};

复杂度分析

• 时间复杂度：$O(nlogn)$。二分查找的时间复杂度为 $O(logn)$，每一轮求得总分球次数的时间复杂度为 $O(n)$。整体时间复杂度为 $O(nlogn)$。
• 空间复杂度：$O(1)$。