题目
数对 (a,b) 由整数 a 和 b 组成，其数对距离定义为 a 和 b 的绝对差值。
给你一个整数数组 nums 和一个整数 k ，数对由 nums[i] 和 nums[j] 组成且满足 0 <= i < j < nums.length 。返回 所有数对距离中 第 k 小的数对距离。
示例
示例 1：
输入：nums = [1,3,1], k = 1
输出：0
解释：数对和对应的距离如下：
(1,3) -> 2
(1,1) -> 0
(3,1) -> 2
距离第 1 小的数对是 (1,1) ，距离为 0 。
示例 2：
输入：nums = [1,1,1], k = 2
输出：0
示例 3：
输入：nums = [1,6,1], k = 3
输出：5
提示：
n == nums.length
2 <= n <= 104
0 <= nums[i] <= 106
1 <= k <= n * (n - 1) / 2
思路
排序+二分+枚举
class Solution(object):
def getCount(self, d, nums, size):
j,count = 0, 0
for i in range(1, size):
while nums[i]-nums[j] > d: j+=1
count += i-j
return count
def smallestDistancePair(self, nums, k):
nums.sort()
size = len(nums)
l = 0
r = nums[size-1] - nums[0]
while l<r:
mid = (l+r) >> 1
count = self.getCount(mid, nums, size)
if count < k: l = mid+1
else: r = mid
return r
|