LeetCode4 两个排序数组的中位数

题目

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。
请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。
你可以假设 nums1 和 nums2 不同时为空。

示例

1. 示例 1:
   nums1 = [1, 3]
   nums2 = [2]
   中位数是 2.0
2. 示例 2:
   nums1 = [1, 2]
   nums2 = [3, 4]
   中位数是 (2 + 3)/2 = 2.5

C++代码

采用了algorithm库中的merge函数，将nums1和nums2合并放入nums3中。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) 
    {
        vector<int> nums3;
        nums3.resize(nums1.size()+nums2.size());
        merge(nums1.begin(),nums1.end(),nums2.begin(),nums2.end(),nums3.begin());
        if(nums3.size()%2!=0)
            return (double) nums3.at((nums3.size()-1)/2);
        else
            return (double)(nums3.at((nums3.size()-1)/2)+nums3.at(nums3.size()/2))/2;
    }
};