[二分查找] 两个有序数组中的第K小数

Posted 热爱生活的小熊猫

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[二分查找] 两个有序数组中的第K小数相关的知识,希望对你有一定的参考价值。

 二分查找系列专题   第四篇

给定两个有序数组arr1和arr2,已知两个数组的⻓度分别为 m1 和 m2,求两个数组中的第 K ⼩数。要求时间复杂度O(log(m1 + m2))。

举例

arr1 = [1,2,3],arr2 = [3,4,5,6],K = 4。
则第 K ⼩数为 3.

arr1 = [0,1,2],arr2 = [3,4,5,7,8], K = 3;
则第 K ⼩数为 2.


题解

本题仍然要求对数时间复杂度,因此也是要使用二分法的,思路和类似。思路:利用二分的思路,每次移动两个数组中左边的指针,把求两个数组的第 小元素不断转化为求解第 小元素,如此缩小范围。

下面给出图解:

int findKth(vector<int>& vec1, int l1, int r1, vector<int>& vec2, int l2, int r2, int k) {
    if (l1 > r1) return vec2[l2 + k];
    if (l2 > r2) return vec1[l1 + k];
    if (k == 0) return min(vec1[l1], vec2[l2]);
    int mid1 = l1 + k / 2 < r1 ? l1 + k / 2 : r1;
    int mid2 = l2 + k / 2 < r2 ? l2 + k / 2 : r2;
    if (vec1[mid1] <= vec2[mid2]) {
        return findKth(vec1, mid1 + 1, r1, vec2, l2, r2, k - k / 2 - 1);
    }
    else {
return findKth(vec1, l1, r1, vec2, mid2 + 1, r2, k - k / 2 - 1);
    }
}

int getKthMin(vector<int>& vec1, vector<int>& vec2, int k) {
    int len1 = vec1.size();
    int len2 = vec2.size();
    if (len1 == 0) return vec2[k - 1];
    if (len2 == 0) return vec1[k - 1];
    return findKth(vec1, 0, len1 - 1, vec2, 0, len2 - 1, k - 1);
}


 二分查找系列往期题目   





优质原创文章来之不易 点个在看给我更新的动力嗷 

以上是关于[二分查找] 两个有序数组中的第K小数的主要内容,如果未能解决你的问题,请参考以下文章

递归打卡2求两个有序数组的第K小数

两个有序数组的第K小乘积

c++ 快排思想查找第k小数……注意是小

LC T668笔记 & 有关二分查找第K小数BFPRT算法

Leetcode4. 寻找两个正序数组的中位数(二分)

二分查找算法复习