LeetCode OJ——Median of Two Sorted Arrays

题目:
There are two sorted arrays nums1 and nums2 of size m and n respectively. Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).


代码:

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        vector<int>::iterator iter1, iter2;
        vector<int> num;  //保存nums1和nums2中共m+n个数据
        int m, n;
        m = nums1.size();
        n = nums2.size();
        iter1 = nums1.begin();
        iter2 = nums2.begin();

        if (m == 0){
            num = nums2;
        }else if (n == 0)
        {
            num = nums1;
        }
        else{
            for (vector<int>::size_type i = 0; i < (m + n); i++)
            {
                if (iter1 != nums1.end() && iter2 != nums2.end())   //nums1, nums2未遍历完
                {
                    if (*iter1 <= *iter2)
                    {
                        num.push_back(*iter1);
                        iter1++;
                    }
                    else{
                        num.push_back(*iter2);
                        iter2++;
                    }
                }
                else if (iter1 == nums1.end())  //nums1遍历完,nums2未遍历完
                {
                    num.insert(num.end(), iter2, nums2.end());
                    break;
                }
                else{   //nums1未遍历完,nums2遍历完
                    num.insert(num.end(), iter1, nums1.end());
                    break;
                }

            }
        }

        if ((m + n) % 2 == 0)   //偶数个数
        {
            return (num[(m + n) / 2 - 1] + num[(m + n ) / 2 ] + 0.0) / 2;
        }
        else{
            return (num[(m + n + 1) / 2 -1] + 0.0);
        }

    }
};

结果:
这里写图片描述


思考:

  • 题意是要找出nums1和nums2中共m+n个数的中位数,那么我最先想到的是要将两个数组按顺序组合成一个新的数组,取新数组的中位数即可

  • 按照上述思想,则代码便能很好理解了。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值