LeetCode刷题系列 -- 719. 找出第 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。
提示:

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-k-th-smallest-pair-distance
 

思路：

 1. 先将数组nums按照从大到小进行排序
 2. 元素之间的最小差值记做 low = 0, 最大差值记做 hi = nums[nums.length-1] - nums[0]
 则 任意两个元素的差值一定在 low 与 hi 之间。
 3. mid = low + (hi - low) / 2.
         1) 定义 left = 0, right = 0, 分别代表子数组的左端下标与右端下标， count 代表满足差小于等于 mid的元素对的数量
         2）遍历 left 到 right 子数组元素，right 从 0开始计算，直到 right == nums.length 为止
             2.1）若是 nums[right] - nums[left] <= mid, 说明一共有 right - left 个元素对满足，count += right - left;
             2.2）若是 nums[right] - nums[left] > mid, 此时 left右移，直到 nums[right] - nums[left] <= mid 为止。
         3）若是  count >= k ，说明差值小于等于 mid 元素对的数量大于等于 k ，则第 k 个最小距离一定小于等于 mid，此时 hi = mid;
 若是  count < k, 说明差值小于等于 mid 元素对的数量小于k ，则第 k 个最小距离一定大于 mid，此时 low = mid + 1;
4. 重复 3 步，直到 low == hi 为止。

java代码如下：

class Solution {
    public int smallestDistancePair(int[] nums, int k) {
        Arrays.sort(nums);

        int low = 0;
        int hi = nums[nums.length - 1] - nums[0];

        while (low < hi) {
            int mid = low + (hi - low) / 2;
            int left = 0;
            int count = 0;
            for (int right = 0; right < nums.length; right++) {
                while (nums[right] - nums[left] > mid) left++;
                count += right - left;
            }
            if (count >= k) {
                hi = mid;
            } else {
                low = mid + 1;
            }
        }
        
        return low;
    }
}