一、问题描述
给你一个整数数组 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
二、思路分析
1.首先,题目要求选择一个正整数作为除数,因此除数应该是大于0的数。
2.由于做除法运算时,都要向上取整,因此当所取整数位数组中最大元素值时,数组中所有整数除以该除数的和为数组元素个数。又因为题目保证一定有解,所以,threshold的最小值应该为数组元素个数。
3.综上分析可知,最小整数的值应该在1至数组最大元素之间。
整体思路
利用二分查找区间[1,maxValue],判断是否存在一个值ret,数组中的所有元素值除以ret的和小于等于threshold并且数组中所有元素的值除以ret-1的和大于threshold。
三、代码描述
class Solution {
public:
//求出数组中元素除以divid的和并返回
int midDivNums(vector<int> nums,int numsSize,int divied)
{
int sum = 0;
for(int i = 0;i < numsSize;i++)
{
if(nums[i]%divied == 0)
{
sum += nums[i]/divied;
}
else
{
sum += nums[i]/divied;
sum++;
}
}
return sum;
}
int smallestDivisor(vector<int>& nums, int threshold) {
//目标值ret应该在[1,maxValue],使用二分查找确定最小的ret
int ret = 1;
//找出数组中的最大值maxValue
int maxValue = nums[0];
int size = nums.size();
for(int i = 0;i < size;i++)
{
if(nums[i] > maxValue)
maxValue = nums[i];
}
//二分查找确定最小的ret
int begin = 1,end = maxValue,mid;
while(begin <= end)
{
mid = (begin+end)/2;
//计算出mid做除数时的总和sum
int sum = midDivNums(nums,size,mid);
//如果sum小于threshold,判断divied是否可以更小
if(sum > threshold)
{
begin = mid+1;
}
else if(sum <= threshold)
{
if(mid == 1 || midDivNums(nums,size,mid-1) > threshold)
{
ret = mid;
break;
}
else
{
end = mid-1;
}
}
}
return ret;
}
};