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.

思路

桶排序

class Solution {
public:
    int smallestDistancePair(vector<int>& nums, int k) {
        int n = nums.size(), N = 1000000;
        vector<int> cnt(N, 0);
        for (int i = 0; i < n; ++i) {
            for (int j = i + 1; j < n; ++j) {
                ++cnt[abs(nums[i] - nums[j])];
            }
        }
        for (int i = 0; i < N; ++i) {
            if (cnt[i] >= k) return i;
            k -= cnt[i];
        }
        return -1;
    }
};

二分法

class Solution {
public:
    int getcount(vector<int>& nums, int mid){
        int ans=0;
        int l=0;
        for(int i=1;i<nums.size();i++){
            while(nums[i]-nums[l]>mid)l++;
            ans+=i-l;
        }
        return ans;
    }
    int smallestDistancePair(vector<int>& nums, int k) {
        sort(nums.begin(),nums.end());
        int len=nums.size(),l=0,r=nums[len-1]-nums[0];
        while(l<r){
            int mid=(l+r)/2;
            if(getcount(nums,mid)<k)l=mid+1;
            else r=mid;
        }
        return l;
    }
};

参考：
桶排序：http://wiki.jikexueyuan.com/project/easy-learn-algorithm/bucket-sort.html
http://www.cnblogs.com/grandyang/p/8627783.html