719.找出第K小的数对距离

6/15

数对 (a,b) 由整数 a 和 b 组成，其数对距离定义为 a 和 b 的绝对差值。

给你一个整数数组 nums 和一个整数 k ，数对由 nums[i] 和 nums[j] 组成且满足 0 <= i < j < nums.length 。返回 所有数对距离中 第 k 小的数对距离。

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

根据解析，主要难点是把距离进行二分查找，根据判断每个距离所拥有的数对数量，来判断该距离是否符合题目要求，值得注意的是，当某些略大于正确结果的距离不存在于数组中的距离时，也符合答案要求，因此要在二分查找中当count和k相等时，应使right=mid-1。

学习到的新函数lower_bound（begin，end，k），返回数组中大于k的第一个数的位置，以指针形式。

代码如下：

int smallestDistancePair(vector<int>& nums, int k) {

        sort(nums.begin(),nums.end());

        int n=nums.size(),left=0,right=nums.back()-nums.front();

        while(left<=right){

            int mid=(left+right)/2;

            int cnt=0;

            for(int j=0;j<n;j++){

                int i=lower_bound(nums.begin(),nums.begin()+j,nums[j]-mid)-nums.begin();

                cnt+=j-i;

            }

            if(cnt>=k){

                right=mid-1;

            }

            else{

                left=mid+1;

            }

        }

        return left;

    }