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

给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

算法的时间复杂度应该为 O(log (m+n)) 。

示例 1:

输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2
示例 2:

输入:nums1 = [1,2], nums2 = [3,4]
输出:2.50000
解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5
示例 3:

输入:nums1 = [0,0], nums2 = [0,0]
输出:0.00000

与合并两个有序数组相似,两个数组不会同时为空,所以如果一个为空直接去另外一个数组找中位数,如果都不为空就比较两个数组的当前元素,那个元素小就合并到新的数组并把数组的指针后移继续比较。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int l1=nums1.size(),l2=nums2.size(),s=l1+l2;
        if(l1==0){
            if(l2%2==1)return nums2[l2/2];
            else return double((nums2[l2/2]+nums2[l2/2-1]))/2;
        }
        if(l2==0){
            if(l1%2==1)return nums1[l1/2];
            else return double((nums1[l1/2]+nums1[l1/2-1]))/2;
        }
        int mid=s/2,flag=1;
        
        vector<int> re;
        if(s%2==0){
            flag=0;
        }
        int count=0,i=0,j=0;
        while(count<=mid){
            if(i<l1&&j<l2){
                if(nums1[i]<nums2[j]){
                    re.push_back(nums1[i]);
                   
                    i++;
                    count++;
                }
                else {
                    re.push_back(nums2[j]);
                    
                    j++;
                    count++;
                }
            }
            else{
                if(j<l2){
                    re.push_back(nums2[j]);
                    
                    j++;
                    count++;
                }
                else {
                    re.push_back(nums1[i]);
                    
                    i++;
                    count++;
                }

            }

        }
        if(flag)return re[mid];
        else return double((re[mid-1]+re[mid]))/2;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值