719. 找出第 k 小的距离对

2021/03/13 leetcode 共 554 字,约 2 分钟

title

problem link

给定一个整数数组,返回所有数对之间的第 k 个最小距离。一对 (A, B) 的距离被定义为 A 和 B 之间的绝对差值。

示例 1:

输入:
nums = [1,3,1]
k = 1
输出:0 
解释:
所有数对如下:
(1,3) -> 2
(1,1) -> 0
(3,1) -> 2
因此第 1 个最小距离的数对是 (1,1),它们之间的距离为 0。

提示:

  1. 2 <= len(nums) <= 10000.
  2. 0 <= nums[i] < 1000000.
  3. 1 <= k <= len(nums) * (len(nums) - 1) / 2.

solution

class Solution:
    def smallestDistancePair(self, nums: List[int], k: int) -> int:
        res = 0
        nums = sorted(nums)
        l, r = 0, nums[-1] - nums[0]

        def count(m):
            ans = 0
            slow = 0
            for i in range(len(nums)):
                while nums[i] - nums[slow] > m:
                    slow += 1
                ans += i-slow
            return ans

        while l <= r:
            mid = (l + r) // 2
            if count(mid) >= k:
                r = mid - 1
            else:
                l = mid + 1
        return l

Search

    Table of Contents