LeetCode #4 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)).

这道题是第一道标为hard的题，要求求出两个已经排序好的数组的中位数，我们知道一个长度为n的排过序的数组A的中位数为A[n/2]（奇数）或者（A[n/2-1]+A[n/2]）/2（偶数），所以只需要把两个数组按照合并排序的方法合并再取下标就好了。当然，也可以不合并，直接在两个数组中从左往右数，每次优先数较小的那个。

下面的提交代码是合并的版本。

提交代码

double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
vector<int> merge;
auto itr1 = nums1.begin();
auto itr2 = nums2.begin();
while(itr1 != nums1.end() || itr2 != nums2.end())
{
if(itr1 == nums1.end())
{
merge.push_back(*itr2++);
}
else if(itr2 == nums2.end())
{
merge.push_back(*itr1++);
}
else
{
merge.push_back(*itr1<*itr2?*itr1++ : *itr2++);
}
}
if(merge.size() & 1)
{
return merge[merge.size() / 2 ];
}
else
{
return (merge[merge.size()/2] + merge[merge.size()/2-1]*1.0)/2;
}
}