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

119 阅读1分钟

要求

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

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

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

题目保证一定有解。

示例 1:

输入:nums = [1,2,5,9], threshold = 6
输出:5
解释:如果除数为 1 ,我们可以得到和为 171+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

核心代码

class Solution:
    def smallestDivisor(self, nums: List[int], threshold: int) -> 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

image.png

解题思路:题目保证有解,而且解的范围非常容易猜到,是[1,max(nums)],所以是典型的二分试探题,所以可以用二分试探法来猜答案,逐步缩小区间长度到1。时间复杂度:O(NlogK),N是nums数组长度,K是max(nums)空间复杂度:O(1)