给你一个整数数组 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, max(nums)】,所以是典型的二分试探题,
所以可以用二分试探法来猜答案,逐步缩小区间长度到1。
时间复杂度:O(NlogK),N是nums数组长度,K是max(nums)
空间复杂度:O(1)
class Solution(object):
def smallestDivisor(self, nums, threshold):
"""
:type nums: List[int]
:type threshold: int
:rtype: int
"""
left, right = 1, max(nums)
while left <= right:
mid = (left + right) // 2
tmp = 0
for num in nums:
tmp += math.ceil(num * 1.0 / mid)
if tmp > threshold:
break
if tmp > threshold:
left = mid + 1
else:
right = mid - 1
return left