leetcode-5281.使结果不超过阈值的最小除数

leetcode-5281.使结果不超过阈值的最小除数(Find the Smallest Divisor Given a Threshold)

题目

给你一个整数数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-the-smallest-divisor-given-a-threshold
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

首先最容易想到的是暴力破解算法，枚举除数k，从k=1开始枚举，每次k++，对每个除数，计算$sum={\Sigma }_{i}ceil(nums[i]/k)$，第一个满足sum<threshold的k即为答案，不过数据量大，暴力算法会超时。

class Solution {
public:
    int smallestDivisor(vector<int>& nums, int threshold) {
        int n = 0;
        int sum = 0;
        do {
            n++;
            sum = 0;
            for (int i = 0; i < nums.size(); i++) {
                sum += ceil(1.0 * nums[i] / n);
            }
        }
        while(sum > threshold);
        return n;
    }
};

设$f(k)={\Sigma }_{i}ceil(nums[i]/k),1<=k<=10^6,k\in N_{+}$，注意到$f(k)$是一个非严格单调递减的离散函数（除数为正整数），$f(k)$有序可以使用二分法快速搜索符合条件的除数。

下面代码使用二分搜索算法查找，和二分查找具体某数略有不同，这里满足f(k) < threshold的k可能有多个，要找最小的那个k。下面代码的注释和图帮助理解这个二分查找的实现细节。

二分查找不断缩小待查找区间，当 l- r == 1时，出现以下三种情况，可以具体分析便于正确实现算法

class Solution {
public:
    int f(int k, vector<int>& nums) {//f(k)，非严格单调递减
        int s = 0;
        for (int i = 0; i < nums.size(); i++) {
            s += ceil(1.0 * nums[i] / k);
        }
        return s;
    }
    
    int smallestDivisor(vector<int>& nums, int threshold) {
        int l = 1, r = 1e6+10;
        int mid = 0;
        while (l < r) {
            mid = (l + r) / 2;
            //if (f(mid, nums) == threshold) return mid;//不能直接返回，可能有多个mid，使得f(mid, nums) == threshold
            if (f(mid, nums) > threshold) {
                l = mid + 1;//f(mid+1)<=threshold，[l,r]任然包含待搜寻答案
            }
            else {
                r = mid;//刚开始写成r=mid-1，可能会跳过待搜索答案
            }
        }
        return l;
    }
};