LeetCode之4. 尋找兩個有序陣列的中位數

1. 給定兩個大小為 m 和 n 的有序陣列 nums1 和 nums2。
   請你找出這兩個有序陣列的中位數，並且要求演算法的時間複雜度為 O(log(m + n))。

2. 你可以假設 nums1 和 nums2 不會同時為空。

示例 1:

nums1 = [1, 3]
nums2 = [2]
則中位數是 2.0

示例 2:

nums1 = [1, 2]
nums2 = [3, 4]
則中位數是 (2 + 3)/2 = 2.5

程式碼如下：

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        multiset<int> s_array;
		double ret = 0.0;

		s_array.insert(nums1.begin(), nums1.end());
		s_array.insert(nums2.begin(), nums2.end());
		vector<int> nums3(s_array.begin(), s_array.end());

		int size_count = s_array.size();
		if (size_count % 2 == 0)
		{
			ret = (nums3.at(size_count / 2) + nums3.at(size_count / 2 - 1)) / 2.0;
		}
		else
		{
			ret = nums3.at((size_count - 1) / 2);
		}

		return ret;
    }
};
 

4. 尋找兩個有序陣列的中位數