4. Median of Two Sorted Arrays

来源:互联网 发布:ubuntu pxe boot 编辑:程序博客网 时间:2024/06/07 00:35

/*

一道比较简单的题目,就是合并两个有序的数组。

*/


class Solution {

public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        vector<int> nums;
        int len1 = nums1.size();
        int len2 = nums2.size();
        int len = len1+len2;
        int cur1=0,cur2=0;
        double res = 0;
        for(int i=0;i<len;i++){
            if(cur1==len1){
                nums.push_back(nums2[cur2]);
                cur2++;
            }
            else if(cur2==len2){
                nums.push_back(nums1[cur1]);
                cur1++;
            }
            else if(nums1[cur1] <= nums2[cur2]){
                nums.push_back(nums1[cur1]);
                cur1++;
            }
            else{
                nums.push_back(nums2[cur2]);
                cur2++;
            }
            if(len%2==0 && i==len/2){
                res = (nums[i]+nums[i-1])/2.0;
            }
            else if(len%2!=0 && i==len/2){
                res = nums[i];
            }
        }
        return res;
    }
};
原创粉丝点击