#4 寻找两个有序数组的中位数(困难)

题目

给定两个大小为 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. 直接组成一个新的有序数组,时间复杂度:O(n1 + n2)
#include
class Solution {
public:
	double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
		double ans = 0.0;
		vector<int> res;
		auto t1 = nums1.begin();
		vector<int>::iterator t2 = nums2.begin();
		while (t1 != nums1.end() && t2 != nums2.end())
		{
			if (*t1 <= *t2)
			{
				res.push_back(*t1);
				t1++;
			}
			else
			{
				res.push_back(*t2);
				t2++;
			}
		}
		if (t1 != nums1.end())
		{
			res.insert(res.end(), t1, nums1.end());
		}
		if (t2 != nums2.end())
		{
			res.insert(res.end(), t2, nums2.end());
		}
		/*while (t1 != nums1.end())
		{
			res.push_back(*t1);
			t1++;
		}*/
		/*while (t2 != nums2.end())
		{
			res.push_back(*t2);
			t2++;
		}*/
		if (res.size() % 2 == 0)
		{
			//偶数
			int a = res.size() / 2;
			ans = (res[a] + res[a - 1]) / 2.0;
		}
		else
		{
			//奇数
			int a = res.size() / 2;
			ans = res[a];
		}

		return ans;
	}
};
  1. 中分法,时间复杂度为:O(log(m + n))
    待更新。。。

你可能感兴趣的:(LeetCode,LeetCode,C++代码)