719. 找出第 K 小的数对距离(二分)

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

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

示例 1:

输入:nums = [1,3,1], k = 1
输出:0
解释:数对和对应的距离如下:
(1,3) -> 2
(1,1) -> 0
(3,1) -> 2
距离第 1 小的数对是 (1,1) ,距离为 0 。
示例 2:

输入:nums = [1,1,1], k = 2
输出:0
示例 3:

输入:nums = [1,6,1], k = 3
输出:5

class Solution {
public:
    int smallestDistancePair(vector<int>& nums, int k) {
        // 排序服务枚举区间
        sort(nums.begin(), nums.end());
        int left = 0, right = nums[nums.size() - 1], n = nums.size();
        while(left < right){
            int mid = (left + right) >> 1, cnt = 0;
            // 枚举区间
            for(int i = 0, j = 0; i < n; ++i){
                // 找到第一个不满足条件的j
                while(j < n && nums[j] - nums[i] <= mid) ++j;
                cnt += j - i - 1;
            }
            // 确定二分方向
            if(cnt < k)
                left = mid + 1;
            else
                right = mid;
        }
        return left;
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值