两个排序数组的中位数
程序员文章站
2024-03-16 12:50:34
...
题目描述
给定两个大小为 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
思路:先对两个子数组合并为一个数据,然后利用冒泡排序对合并后的数据进行排序。求中位数,如果是偶数个数,中位数就是中间两个数的和除以2;如果是奇数个数,中位数就是中间的那个数。
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
void bubble(int *a,int len);
int len = nums1Size + nums2Size;
int myArray[1000];
int *p;
double mid = -1.0;
p = myArray;
int i,j;
for(i=0;i<nums1Size;i++)
{
*(p+i) = *nums1++;
}
for(i = nums1Size-1,j=0;i<nums1Size + nums2Size;i++,j++)
{
*(p+i) = *nums2;
}
bubble(*p,len);
if(len%2==0)
mid = 1.0*(*(p+(len-1)/2 )+ *(p + (len-1)/2 + 1))/2;
else
mid = 1.0*(*(p + (len-1)/2));
return mid;
}
void bubble(int *a,int len)
{
int i,j;
int temp;
int *pt;
pt = a;
for(i=0;i<len-1;i++)
{
for(j=0;j<len-1-i;j++)
{
if(*(pt+j)>*(pt+j+1))
{
temp = *(pt+j+1);
*(pt+j+1) = *(pt+j);
*(pt+j) = temp;
}
}
}
}
下一篇: 需要排序的最短子数组长度