Leetcode 719. 找出第 k 小的距离对 C++

Leetcode 719. 找出第 k 小的距离对

题目

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

示例:

输入：
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.

题解

二分查找+滑动窗口
我们将nums排序后，显然相邻的数对的绝对值相差要小。因此，如果我们规定一个阈值，找小于等于阈值的数对个数，可以使用滑动窗口的方法。
我们使每个数作为我们的窗口左边界，找到尽可能大的窗口右边界，使最大数-最小数小于等于阈值的个数，因此我们就能知道所有小于等于阈值的数对数。
显然，阈值设的越大，数对个数越多，因此我们可以使用二分查找的思想。最小阈值肯定是0，最大阈值则为最大数-最小数。详细过程见代码

代码

	bool isBiggerK(vector<int>& nums,int k,int now){		//数对差小于等于now的个数是否大于k
        int i=0,j=0;
        int ans = 0;
        while(i < nums.size()){
            if(j < i)   j=i;
            while(j<nums.size() && nums[j]-nums[i] <= now){
                j++;
            }
            if(j!=i) ans += j-i-1;
            if(ans >= k) return true;
            i++;
        }
        return false;
    }
    int smallestDistancePair(vector<int>& nums, int k) {
        sort(nums.begin(),nums.end());
        int n = nums.size();
        int l=0,r=nums[n-1]-nums[0],mid;
        while(l < r){
            mid = (l+r)/2;
            if(!isBiggerK(nums,k,mid))		//小于k，我们需要增大阈值
                l = mid+1;
            else			//否则，则减小阈值
                r = mid;
        }
        return l;
    }

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