LeetCode系列5：尋找兩個有序數組的中位數

題目

給定兩個大小爲 m 和 n 的有序數組 nums1 和 nums2。

請你找出這兩個有序數組的中位數，並且要求算法的時間複雜度爲 O(log(m + n))。

你可以假設 nums1 和 nums2 不會同時爲空。

示例 1:

nums1 = [1, 3]
nums2 = [2]

則中位數是 2.0
示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

則中位數是 (2 + 3)/2 = 2.5

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/median-of-two-sorted-arrays
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

解題思路

1. 將兩個數組合並
2. 排序
3. 計算中位數
4. 注意數據類型轉換

實現源碼

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        vector<int> v;
		v.resize(nums1.size() + nums2.size());
		// merge 默認會將數組排序
		merge(nums1.begin(), nums1.end(), nums2.begin(), nums2.end(), v.begin());
		if (v.size() % 2 == 0)
		{ 
			return (double)((float)(v.at(v.size() / 2) + v.at(v.size() / 2 - 1))) / 2;
		}
		else
		{
			return v.at(v.size() / 2);
		} 
		return 0;
    }
};