c_cpp 两个排序数组的中位数的.cpp

Posted 2021-05-14

有两个大小为 m 和 n 的排序数组 nums1 和 nums2 。
请找出两个排序数组的中位数并且总的运行时间复杂度为 O(log (m+n)) 。
示例 1:
nums1 = [1, 3]
nums2 = [2]
中位数是 2.0

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


vector<int> MySort(vector<int> &nums1,vector<int> &nums2)
{
	vector<int> vec_re;
	auto it1 = nums1.begin();
	auto it2 = nums2.begin();
	while(it1 != nums1.end() && it2 != nums2.end())
	{
		if(*it1 < *it2)
		{
			vec_re.push_back(*it1);
			it1++;nums1
		}
		else
		{
			vec_re.push_back(*it2);
			it2++;
		}

	}
	if(it1 != nums1.end())
	{
		while(it1 != nums1.end())
		{
			vec_re.push_back(*it1);
			it1++;
		}
	}
	if(it2 != nums2.end())
	{
		while(it2 != nums2.end())
		{
			vec_re.push_back(*it2);
			it2++;
		}	
	}
	
	return vec_re;
}



double findMedianSortedArrays(vector<int> &nums1,vector<int> &nums2)
{	
	int size1 = 0, size2 = 0;
	size1 = nums1.size();
	size2 = nums2.size();
	
	
	int index = 0, median_index = 0, temp;
	auto it_index = find(nums1.begin(), nums1.end(), nums2[0]);
	
	index = ceil( (size1+size2)/2 ) - 1;
	vector<int> vec_new = MySort(nums1, nums2);
	index = ceil( (size1+size2)/2 );
	
	if((size1+size2)%2)
	{
		return vec_new[index];
	}
	else {
		return (double)(vec_new[index] + vec_new[index - 1])/2;
	}
}