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