每日一题·719.找出第k个小的数对距离

最新推荐文章于 2022-07-13 15:33:12 发布

迅~

最新推荐文章于 2022-07-13 15:33:12 发布

阅读量207

点赞数

分类专栏： LeetCode刷题笔记文章标签：算法 c语言数据结构

本文链接：https://blog.csdn.net/m0_64560763/article/details/125305698

版权

LeetCode刷题笔记专栏收录该内容

440 篇文章 4 订阅

订阅专栏

题目

数对 (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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

示例

思路

先对数组进行排序，所以第k个数对距离肯定在max(nums) - min(nums)之中

每次二分查找目标值，比较在nums中有多少对数小于目标值sum

如果sum小于k，说明目标值小了，所以收缩左边

如果sum大于等于k，说明目标值大了，所以收缩右边

为什么等于与收缩右边而不是直接返回退出

因为我们是在max(nums) - min(nums)之中找的目标值，这个目标值不一定在nums数对距离之中，所以需要不断比较，找到等于k，并且又在nums数对距离中存在的值

代码

int cmp(const void * a , const void * b)
{
    return *(int *)a - *(int *)b;
}
/*
*smallestDistancePair:在数组中找出第k个小的数对距离
int* nums：给定数组
int numsSize：数组长度
int k：第k小的值
*/
int smallestDistancePair(int* nums, int numsSize, int k) {
    qsort(nums , numsSize , sizeof(nums[0]) , cmp);
    int i , j ;
    int sum;
    int mid;
    int left = 0 ,right = nums[numsSize-1] - nums[0];
    while(left <= right)
    {
        sum = 0;
        mid = (left + right) / 2;
        for(i = 0 , j = 0; j < numsSize; j++)
        {
            while(nums[j] - nums[i] > mid)
            {
                i++;
            }
            sum += j - i;
        }
        if(sum >= k)
        {
            right = mid - 1;
        }
        else
        {
            left = mid + 1;
        }
    }
    return left;
}