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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果：

代码描述：

class Solution {
public:
    int smallestDistancePair(vector<int>& nums, int k) {
        // 先对数组进行排序
        sort(nums.begin(), nums.end());
        int len = nums.size();
        int left = 0, right = nums[len-1]-nums[0];
        int mid;    // 中间值，并非中间索引
        int count;
        while(left < right)
        {
            mid = left + (right - left)/2;
            count = getCount(nums, mid);
            //注意这里的二分查找左边界和右边界更新的标准
            //小于k，mid必然非最终解，因此left = mid + 1
            //大于等于k，可以将mid作为候选解
            if(count < k)
                left = mid + 1;
            else
                right = mid;
        }
        return left;
    }
    // 计数方式
    int getCount(vector<int> &nums, int mid)
    {
        int count = 0;
        int left = 0;
        //注意这里的left，在迭代的过程中没有重置，而是一直右移
        //很容易理解，i右移，nums[i]变大(注意对数组进行过排序)，而nums[left](上一轮的)不变
        //因此差增大，必然还是比mid大，因此不需要再迭代一遍
        for(int i = 1; i < nums.size(); ++i)
        {
            while(nums[i] - nums[left] > mid)
                ++left;
            count += i - left;
        }
        return count;
    }
};