LeetCode - 719. 找出第 k 小的距离对

719. 找出第 k 小的距离对

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

解题思路： 利用以前解TOP-K的方法解此题，比如用堆，OJ会TLE，因为算法的时间复杂度达到$O(N^{2}logN)$，此处利用桶排序解题，算法复杂度为$O(N^2)$，刚好可以通过OJ.

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

2. 解法2，二分查找

class Solution {
public:
    int smallestDistancePair(vector<int>& nums, int k) {
        sort(nums.begin(), nums.end());
        int n = nums.size();
        int left = 0, right = nums.back() - nums[0];
        while (left < right) {
            int cnt = 0, start = 0, mid = (left + right) / 2;
            for (int i = 0; i < n; ++i) {
                while (start < n && nums[i] - nums[start] > mid) ++start;
                cnt += (i - start);
            }
            if (cnt < k) left = mid + 1;
            else right = mid;
        }
        return right;
    }
};