【LeetCode】719. 找出第 k 小的距离对

前言

找出第 k 小的距离对
给定一个整数数组，返回所有数对之间的第 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。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-k-th-smallest-pair-distance
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

找出第k小的距离对，可用二分查找
1、数组排序
2、最小距离对 = 0，最大距离对 = nums[len - 1] - nums[0]
3、第k小距离对就在最小距离对 和 最大距离对之间

/**
 * 二分查找 + 双指针
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var smallestDistancePair = function(nums, k) {
  nums.sort((a, b) => a - b)
  const len = nums.length
  let left = 0, right = nums[len - 1] - nums[0]
  while (left < right) {
    const mid = Math.floor(left + (right - left) / 2)
    if (getCount(mid, nums) >= k) {
      right = mid
    } else {
      left = mid + 1
    }
  }
  return right
};

var getCount = function(mid, nums) {
  let l = 0, c = 0
  for (let r = 1; r < nums.length; r++) {
    while (nums[r] - nums[l] > mid) {
      l++
    }
    c += r - l
  }
  return c
}

总结

二分查找的具体应用