两个有序数组求第k小元素

这篇博客介绍了如何利用二分查找算法解决在两个已排序数组中寻找第k小元素的问题。通过在其中一个数组中进行二分搜索,根据搜索过程中的元素比较更新边界,最终确定第k小元素的位置。这种方法的关键在于巧妙地结合两个数组的索引关系,确保在搜索过程中逐步逼近目标值。
摘要由CSDN通过智能技术生成

和两个数组求中位数有点类似. 思路还是二分查找, 也就是在一个数组中进行二分搜索, 每次比较中点位置元素和另一个数组对应位置的大小关系. 比如要找第K个数, 并且当前搜索到了数组A的x位置, 那么要比较的B数组中的位置应该为(k-x-1), 也就是使得数组A中到索引x的元素个数+B中到(k-x-2)索引的元素个数之和为k, 也就是:

A:   x            x+1

B:   k-x-2     k-x-1

对于数组A中的x索引, 比较A[x] 和 B[k-x-1], 如果A[x] < B[k-x-1]则移动left = mid+1, 否则right = mid-1. 因为索引x之前(包括x)有x+1个元素, 索引(k-x-2)之前有(k-x-1)个元素, 这样的话如果A[x] > B[k-x-1]的话说明A[x]必然是属于第k个以后的数; 如果A[x] < B[k-x-1], 说明x索引还需要继续向右移动. 这样最终right指针会停留在x位置, left指针会停留在x+1位置上, 我们要找到第k个数只需要比较A[x]和B[k-x-2]即可.

代码如下:

    int findKth(vector<int>& nums1, vector<int>& nums2, int k)
    {
        if(nums1.size() > nums2.size()) swap(nums1, nums2);
        int m = min(k, (int)nums1.size()), n = min(k, (int)nums2.size());
        int left = 0, right = m-1;
        while(left <= right)
        {
            int mid = left + (right-left)/2, index = k-mid-1;
            if(index >= n || nums1[mid] < nums2[index]) left = mid+1;
            else right = mid-1;
        }
        return max(right<0?INT_MIN:nums1[right], nums2[k-right-2]);
    }


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值