【LeetCode4.寻找两个正序数组的中位数】二分O(log(m+n))

题目链接

4. 寻找两个正序数组的中位数 - 力扣(LeetCode)

实现思路

  • **本质上,其实就是求两个数组第k大的数的变式。**设两个数组大小为m、n,则总长度为t=m+n,那么如果t为奇数,求的就是两个数组第t/2+1大的数;如果t为偶数,求的就是两个数组第t/2和第t/2+1大的数的平均数。
  • 那么,求两个数组第k大的数,每次判断当前数组第k/2大的数的大小关系,如果数组1对应的位置aj小于数组2对应位置bj的元素,那么aj连同之前的数字,必然不可能是第k大的数!这样的话,就移动数组1的开始下标,并且k减去k/2,再次计算当前开始下标的条件下,两个数组第k(其实就是上一次的k-k/2)大的元素。

可以大致看一下下面的图解:

代码实现

cpp 复制代码
class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m = nums1.size();
        int n = nums2.size();
        int t = m + n;
        if (t % 2) {
            return find(nums1, 0, m - 1, nums2, 0, n - 1, t / 2 + 1);
        } else {
            return (find(nums1, 0, m - 1, nums2, 0, n - 1, t / 2) + 
                find(nums1, 0, m - 1, nums2, 0, n - 1, t / 2 + 1)) / 2.0;
        }
    }
    // 找到两个数组中合并后第k大的数
    int find(vector<int>& nums1, int astart, int aend, vector<int>& nums2, int bstart, int bend, int k) {
        if (astart > aend) {
            return nums2[bstart + k - 1];
        }
        if (bstart > bend) {
            return nums1[astart + k - 1];
        }
        if (k == 1) {
            return min(nums1[astart], nums2[bstart]);
        }
        int q = k / 2;
        // 注意边界判断
        int aj = min(astart + q - 1, aend);
        int bj = min(bstart + q - 1, bend);
        if (nums1[aj] <= nums2[bj]) {
            return find(nums1, aj + 1, aend, nums2, bstart, bend, k - (aj - astart + 1));
        } else {
            return find(nums1, astart, aend, nums2, bj + 1, bend, k - (bj - bstart + 1));
        }
    }
};
相关推荐
AA陈超28 分钟前
虚幻引擎UE5专用服务器游戏开发-20 添加基础能力类与连招能力
c++·游戏·ue5·游戏引擎·虚幻
mit6.8241 小时前
[Meetily后端框架] AI摘要结构化 | `SummaryResponse`模型 | Pydantic库 | vs marshmallow库
c++·人工智能·后端
R-G-B1 小时前
【02】MFC入门到精通——MFC 手动添加创建新的对话框模板
c++·mfc·mfc 手动添加创建新的对话框
linux kernel1 小时前
第七讲:C++中的string类
开发语言·c++
jz_ddk1 小时前
[实战]调频(FM)和调幅(AM)信号生成(完整C语言实现)
c语言·算法·信号处理
Tipriest_1 小时前
[数据结构与算法] 优先队列 | 最小堆 C++
c++·优先队列·数据结构与算法·最小堆
CloudAce云一1 小时前
谷歌云代理商:谷歌云TPU/GPU如何加速您的AI模型训练和推理
算法
宛西南浪漫戈命2 小时前
Centos 7下使用C++使用Rdkafka库实现生产者消费者
c++·centos·linq
轻语呢喃2 小时前
每日LeetCode : 杨辉三角
javascript·后端·算法
YuTaoShao3 小时前
【LeetCode 热题 100】148. 排序链表——(解法二)分治
java·算法·leetcode·链表