【二分查找】leetcode 1283. 使结果不超过阈值的最小除数

lele_ne

于 2022-08-09 14:03:06 发布

阅读量118

点赞数

分类专栏： # 二分查找算法之路文章标签： leetcode 算法

本文链接：https://blog.csdn.net/lele_ne/article/details/126245987

版权

算法之路同时被 2 个专栏收录

84 篇文章 0 订阅

订阅专栏

二分查找

26 篇文章 0 订阅

订阅专栏

1283. 使结果不超过阈值的最小除数

题目描述

给你一个整数数组 nums 和一个正整数 threshold ，你需要选择一个正整数作为除数，然后将数组里每个数都除以它，并对除法结果求和。

请你找出能够使上述结果小于等于阈值 threshold 的除数中最小的那个。

每个数除以除数后都向上取整，比方说 7/3 = 3 ， 10/2 = 5 。

题目保证一定有解。

示例1：

输入： nums = [1,2,5,9], threshold = 6
输出： 5
解释： 如果除数为 1 ，我们可以得到和为 17 （1+2+5+9）。
如果除数为 4 ，我们可以得到和为 7 (1+1+2+3) 。如果除数为 5 ，和为 5 (1+1+1+2)。

示例2：

输入： nums = [2,3,5,7,11], threshold = 11
输出： 3

示例3：

输入： nums = [19], threshold = 5
输出： 4

提示

$1 <= nums.length <= 5 * 10^4$
$1 <= nums[i] <= 10^6$
$nums.length <= threshold <= 10^6$

方法：二分查找

解题思路

由题意得，当除数越大时，数组每个元素除以它得到的结果总和越小；当除数越小时，数组每个元素除以它得到的结果总和越大。因此，本题具有单调性，可以利用二分查找来解题。

二分查找的左边界为 1，因为除数不能为 0；右边界为数组中的最大值，这样每个元素除以除数的结果均为1，总和为数组的长度 $n u m s . l e n g t h$ 。

如果 $s u m <= t h res h o l d$ ，则将区间缩小为 $[l, mi d]$ ；
如果 $s u m > t h res h o l d$ ，则将区间缩小为 $[mi d + 1, r]$ ；

直到找到使结果总和不超过 $t h res h o l d$ 的情况下，除数取最小。

代码

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