【LeetCode热题100】【二分查找】寻找两个正序数组的中位数

题目链接:4. 寻找两个正序数组的中位数 - 力扣(LeetCode)

在两个有序数组中在不合并的前提下寻找他们合并后的中位数,这个可以转换成寻找第k大的数

我们来看看这个第k大的数应该在什么地方,下标从0开始,那么a[i]前面有i个数,b[j]前面有j个数

那么如果i+j=k,那么说明a[i]和b[j]前面有k个数,如果a[i-1]<=b[j]并且b[j-1]<=a[i],这就说明这k个数就是合并后有序的前k个数,那么第k大的数就是a[i-1]和b[j-1]之间的较大者

所以可以在a中二分查找符合条件的i,让j=k-i,如果b[j-1]>a[i]说明i的位置应该往后挪查找,否则往前查找

这里需要注意j的范围,因为b[j-1]和b[j]可以不存在,比如a=123,b=456,如果要找3,那么i=3,j=0,此时b[i-1]是不存在的,a[i]也是不存在的,因此0<=j<=m,即0<=k-i<=m,即i<=k,i>=k-m

这样可以二分确定第k大的范围,那么要找中位数,如果总长度是偶数,那么中位数就是中间两个数的平均值,否则就是中间那个数

复制代码
class Solution {
public:
    int findK(vector<int> &nums1, vector<int> &nums2, int k) {
        int left = max(0, int(k - nums2.size())), right = min(k, int(nums1.size()));
        while (left < right) {
            int mid = (left + right) / 2;
            if (nums2[k - mid - 1] > nums1[mid])
                left = mid + 1;
            else
                right = mid;
        }
        return max(left ? nums1[left - 1] : INT_MIN, k - left ? nums2[k - left - 1] : INT_MIN);
    }

    double findMedianSortedArrays(vector<int> &nums1, vector<int> &nums2) {
        int mn = nums1.size() + nums2.size();
        if (mn & 1)
            return findK(nums1, nums2, mn / 2 + 1);
        return (findK(nums1, nums2, mn / 2) + findK(nums1, nums2, mn / 2 + 1)) / 2.0;
    }
};
相关推荐
CoovallyAIHub3 分钟前
YOLO-Maste开源:首个MoE加速加速实时检测,推理提速17.8%!
深度学习·算法·计算机视觉
清铎6 分钟前
leetcode_day13_普通数组_《绝境求生》
数据结构·算法
hetao173383717 分钟前
2026-01-09~12 hetao1733837 的刷题笔记
c++·笔记·算法
过河卒_zh15667661 小时前
情感型AI被“立规矩”,AI陪伴时代进入下半场
人工智能·算法·aigc·生成式人工智能·算法备案
wefg11 小时前
【算法】动态规划
算法·动态规划
机器学习之心1 小时前
198种组合算法+优化TCN-Transformer+SHAP分析+新数据预测+多输出!深度学习可解释分析,强烈安利,粉丝必备!
深度学习·算法·transformer·shap分析·新数据预测
狐571 小时前
2026-01-12-LeetCode刷题笔记-1266-访问所有点的最小时间.md
笔记·算法·leetcode
Gorgous—l1 小时前
数据结构算法学习:LeetCode热题100-栈篇(有效的括号、最小栈、字符串解码、每日温度、柱状图中最大的矩形)
数据结构·学习·算法
小郭团队1 小时前
教育公平的探索
大数据·人工智能·嵌入式硬件·算法·硬件架构
瑞雨溪1 小时前
力扣题解:740.删除并获得点数
算法·leetcode·职场和发展