题目:
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
要求:算法的时间复杂度应该为 O(log (m+n)) 。
算法思想:
一、使用最通俗易懂的方法:
1,先将nums1和nums2合并成一个有序数组arr
2,再将arr求取中位数
代码如下(c语言):
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size)
{
int p1 = 0, p2 = 0;
int m = 0,l = 0,k = 0;
int arr[nums1Size + nums2Size];//新建一个总长度为nums1Size + nums2Size的数组
int q;
double n;
while (p1 < nums1Size || p2 < nums2Size)
{
if (p1 == nums1Size)
{
q = nums2[p2++];
}
else if (p2 == nums2Size)
{
q = nums1[p1++];
}
else if (nums1[p1] < nums2[p2])
{
q = nums1[p1++];
}
else
{
q = nums2[p2++];
}
arr[p1 + p2 - 1] = q;
}
m = nums1Size + nums2Size;
if(m == 1) return arr[m-1];
else
{
if(m%2 == 0)
{
n = (arr[m/2] + arr[m/2-1])/2.0;
}
else
{
n = arr[m/2];
}
}
return n;
}
二、使用暴力插入 + 冒泡
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){
int numsize=nums1Size+nums2Size;//获取两个数组的总长度
int nums[numsize];//创建总长度大小的数组
int n=numsize/2,i,t,j;//n是数组的中位置,后面用于分类
//将第一个数组插入到大数组中
for(i=0;i<nums1Size;i++)
{
nums[i]=nums1[i];
}
//将第二个数组插入到大数组中
for(i=0;i<nums2Size;i++)//两个for暴力入数组
{
nums[i+nums1Size]=nums2[i];//温馨提示,自己写的时候注意下标
}
//使用双重for循环,定义快慢索引,进行冒泡排序
for(i=0;i<numsize;i++)
{
for(j=i+1;j<numsize;j++)
{
if(nums[i]>nums[j])//最小的在第一位,升序
{
//进行交换
t=nums[i];
nums[i]=nums[j];
nums[j]=t;
}
}
}
//判断中位数
if(numsize%2==1) return nums[n];//如果总数有奇数个,中位数就是中间那个
else return (nums[n-1]+nums[n])/2.0;//如果是偶数个,两个数相加除以2
}