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

题目:
给定两个大小分别为 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

思路:
根据替题目提干,以及示例的提示可知,题目的目的是将两个有序数组合并为一个有序数组后,找到新新数组的中位数。中位数在数组中的index 取决于数组的长度的奇偶性。
int size = 新数组.size();
size == 偶数: 中位数的index:
int n = size/2;
index = n + n-1;

size == 奇数:中位数的index:
index = size/2

//实现一:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
    int size1 = nums1.size();
    int size2 = nums2.size();
    std::vector<int> vResult;
    int i = 0, j = 0;
    for(int count = 0 ;count < size1+size2; count++)
    {
        if(i< size1 && j < size2)
        {
            if(nums1[i] < nums2[j])
            {
                vResult.push_back(nums1[i]);
                i++;
            }
            else if(nums1[i] > nums2[j])
            {
                vResult.push_back(nums2[j]);
                j++;
            }
            else
            {
                vResult.push_back(nums1[i]);
                vResult.push_back(nums2[j]);
                j++;
                i++;
            }
        }
        else if(j < size2 && i == size1)
        {
            for(;j<size2;j++)
            {
                vResult.push_back(nums2[j]);
            }
        }
        else if(i < size1 && j == size2)
        {
            for(;i<size1;i++)
            {
                vResult.push_back(nums1[i]);
            }
        }
    }
    int sizeResult = vResult.size();
    if(sizeResult % 2 != 0)
    {
        return  static_cast<float>(vResult[(vResult.size()-1)/2]);
    }
    else
    {
        int n = sizeResult/2;
        return static_cast<float>((vResult[n] + vResult[n-1])/2.0);
    }
}
//实现二: 调用vector 的库函数 sort来实现vector中数据排序
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
    for(unsigned i = 0;i < nums2.size();i++)
    {
        nums1.push_back(nums2[i]);
    }
    sort(nums1.begin(),nums1.end());
    int sizeResult = nums1.size();
    float out;
    if(sizeResult % 2 != 0)
    {
        out =   static_cast<float>(nums1[(sizeResult-1)/2]);
    }
    else
    {
        int n = sizeResult/2;
        out =  static_cast<float>((nums1[n] + nums1[n-1])/2.0);
    }
    return out;
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

xinran0703

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值