LeetCode04：寻找中位数

给定两个大小为 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


思路：这道题也比较简单，将两个有序数组进行合并成一个有序数组即可。

#pragma once
#include <vector>
using namespace std;

/*
执行用时 : 116 ms, 在Median of Two Sorted Arrays的C提交中击败了3.98% 的用户
内存消耗 : 49.1 MB, 在Median of Two Sorted Arrays的C提交中击败了0.00% 的用户
*/
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
	int k = 0, j = 0;
	double mid = 0;
	int length = nums1Size + nums2Size;
	int *num = (int *)malloc(10 * sizeof(int));

	for (int i = 0; i < nums1Size + nums2Size; i++)
	{
		if (k < nums1Size && j < nums2Size)
		{
			int temp = nums1[k] < nums2[j] ? nums1[k] : nums2[j];
			num[i] = temp;
			if (temp == nums1[k])
				k++;
			else
				j++;
		}
		else if( k < nums1Size)
		{
			num[i] = nums1[k];
			k++;
		}
		else
		{
			num[i] = nums2[j];
			j++;
		}
	}
	if (length % 2 == 0)
		mid = (num[length / 2] + num[length / 2 - 1]) / 2.0;
	else
		mid = num[length / 2];
	free(num);
	return mid;
}