给定两个大小分别为 m
和 n
的正序(从小到大)数组 nums1
和 nums2
。请你找出并返回这两个正序数组的 中位数 。
算法的时间复杂度应该为 O(log (m+n))
。
这道题要求合并数组,找到中位数,但是要求时间复杂度在O(log(m+n)),所以我们在合并过程中就要找到中位数,不需要在合并之后再去遍历数组
我们在合并数组的过程中如果将数组已经排好顺序,我们就可以找到中位数,不需要去遍历数组
这里我们可以想到归并排序
归并排序是利用分治算法把一个数组分成一个个有序的数组,然后两两归并
我们这里可以把两个数组看作要归并的一组,套用归并部分即可
我们先开辟一块空间
设置两个变量p、q,分别从两个数组的第一个元素开始遍历数组,从第一个数字开始比较,把较小p对应的数字拷贝到新数组中,然后这个变量p向后遍历一个位置,再与q作比较,一直到其中一个数组全部遍历完,另一个数组剩下的数字就是比前面所有数字都大的正序排列的数了,我们把这些数字按序全部拷贝到新数组的后面。
然后从新数组中找中位数,把找到的中位数强制转换成double型
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
//1.先合并数组
//2.返回中位数
//合并数组
//这里利用归并排序中的将分治数组归并的算法
int length = nums1Size + nums2Size;
int* num = (int*)malloc(sizeof(int) * length);
if (num == NULL)
{
printf("malloc failed\n");
exit(1);
}
int i = 0;
int p1 = i;
int p2 = i;
while (p1 < nums1Size&&p2<nums2Size)
{
if (nums1[p1] <=nums2[p2])
{
num[i++] = nums1[p1++];
}
else
{
num[i++] = nums2[p2++];
}
}
if (p1>=nums1Size)
{
for (int j = p2; j < nums2Size; j++)
{
num[i++] = nums2[p2++];
}
}
if(p2>=nums2Size)
{
for (int j = p1; j < nums1Size; j++)
{
num[i++] = nums1[p1++];
}
}
// if (length % 2 == 1)
// {
// return (double)num[length / 2];
// }
// else if (length % 2 == 0)
// {
// return (double)(num[length / 2] + num[length / 2 - 1]) / 2;
// }
if (length % 2 == 1) {
return (double)num[length / 2];
} else if (length % 2 == 0) {
return (double)(num[length / 2] + num[length / 2 - 1]) / 2;
}
// Add a default return statement
return 0.0; // or any other appropriate value
}