4. 寻找两个正序数组的中位数

题目描述

2个有序数组(保证不能同时为空)长度分别为m,n;求他们的中位数。

要求时间复杂度O(long(m+n))。

解题思路

题目的要求可以转述为求第k大个数,k可能为1个数,可能为2个数。
k=(m+n)/2

  1. num1[k/2]表示第一个数组的第k/2个数。
  2. num2[k/2]表示第二个数组的第k/2个数。

下面进行分类讨论:

  • num1[k/2]<num2[k/2],那么只需要比较num1(k/2+1,m)和num2(0,n),且去掉了k/2个数。
  • num1[k/2]>num2[k/2],那么只需要比较num2(k/2+1,n)和num1(0,m),且去掉了k/2个数。
  • num1[k/2]=num2[k/2],该值就是答案。

每一次的比较k要进行修改。因为已经排除了k/2个数。

上述讨论没有结果就继续执行(递归或者循环)。

注意:

  • m+n为奇数,中位数1个。
  • m+n为偶数,中位数2个。

1个就找1遍,2个就找 2遍即可。

细节问题看具体的代码实现。

代码

cpp 复制代码
class Solution {

public:

    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {

        int m=nums1.size(),n=nums2.size();

        if((m+n)%2==0)

        {

            return (findMedian(nums1,nums2,0,0,(m+n)/2)+findMedian(nums1,nums2,0,0,(m+n)/2+1))/2.0;

        }

        else

        {

            return findMedian(nums1,nums2,0,0,(m+n)/2+1);

        }

    }

    int findMedian(vector<int>& nums1, vector<int>& nums2,int sp1,int sp2,int k)

    {

        //保证nums1中要查找的元素个数小于nums2中的,是为了方便对边界进行处理。

        if(nums1.size()-sp1>nums2.size()-sp2) return findMedian(nums2,nums1,sp2,sp1,k);

  

        //当k位1的时候,就是两个数组中第一个元素中的最小值

        if(k==1)

        {

            if(nums1.size()==sp1) return nums2[sp2];

            return min(nums1[sp1],nums2[sp2]);

        }

        //因为第一个数组查找的数据较小 ,容易越界

        if(nums1.size()==sp1) return nums2[sp2+k-1];

        //第一个数组,当越界的时候,只需要比较较小值即可。

        int i=min(sp1+k/2-1,(int)nums1.size()-1),j=sp2+k-k/2-1;

  

        if(nums1[i]<nums2[j]) return findMedian(nums1,nums2,i+1,sp2,k-(i-sp1+1));

        else  return findMedian(nums1,nums2,sp1,j+1,k-(j-sp2+1));

    }

};
相关推荐
xxwl5852 分钟前
区域赛算法模版第八章---贪心(Greedy)
c++·笔记·学习·算法
CCblandy42 分钟前
向量检索 ANN 参数实测:换一种数据分布,recall 掉了 51 个百分点
人工智能·python·算法·自然语言处理·性能优化
CS创新实验室42 分钟前
一棵树怎么做到“自平衡还不太费力“?——红黑树的旋转与染色
数据结构·算法·红黑树·408·计算机考研
Logic1011 小时前
C语言/数据结构位运算题解:异或XOR找出被挑选一次的果子编号——只出现一次的数字
c语言·数据结构·算法·数组·位运算·时间复杂度·空间复杂度
深圳市快瞳科技有限公司1 小时前
面向端侧部署的多帧宠物身份识别:完整性评分机制与工程约束
人工智能·算法·计算机视觉·端侧ai·宠物识别·宠物ai·宠物身份识别
从小白到软工3 小时前
4-3 项目-学生成绩管理系统(上)
c语言·开发语言·数据结构·经验分享·ai编程
悠仁さん3 小时前
非类型模板参数的特化与偏特化
java·数据结构·算法
垆边人似月.3 小时前
最优河堤加固(二分答案 + 前缀和)
算法
171320330计算机毕设编程3 小时前
计算机毕业设计 基于Hadoop的新能源汽车发展前景分析 LW PPT
java·大数据·hadoop·python·算法·随机森林·线性回归
haluhalu.3 小时前
深入理解 AI Agent(1.2):Harness 工程——模型之外的竞争力
前端·数据结构·人工智能·python·langchain·个人开发