题目
给定两个大小为 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
代码
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){
double temp[100000];
int l=0,k=0,num=0,i=0;
if(nums1Size>=nums2Size)
{ for(i=0;i<nums1Size;i++)
{ if(nums2Size<=l||nums1[i]<nums2[l])
{
temp[k++]=nums1[i];
num++;
}
else
{
temp[k++]=nums2[l++];
num++;
i--;
}
if(num==(nums1Size+nums2Size)/2+1)
break;
}
if(i==nums1Size&&num<(nums1Size+nums2Size)/2+1)
{
while(num<(nums1Size+nums2Size)/2+1){
temp[k++]=nums2[l++];
num++;
}
}
}
else
for(i=0;i<nums2Size;i++)
{
if(nums1Size<=l||nums2[i]<nums1[l])
{
temp[k++]=nums2[i];
num++;
}
else
{
temp[k++]=nums1[l++];
num++;
i--;
}
if(num==(nums1Size+nums2Size)/2+1)
break;
if(i==nums2Size&&num<(nums1Size+nums2Size)/2+1)
while(num<(nums1Size+nums2Size)/2+1){
temp[k++]=nums1[l++];
num++;
}
}
if(num>=1)
if((nums1Size+nums2Size)%2!=0)
return temp[num-1];
else
return (temp[num-2]+temp[num-1])/2;
else
return temp[0];
}
主要的思路就是,创建temp数组,按顺序存储两个数组的和,但是由于只要找中位数,所以temp数组主需要存储到两个数组的一半就可以了,例如:
nums1:1 2 4 6
nums2:3 5 7
temp:1 2 3 4
因为num1和num2的个数总和为7,所以temp只需要存储到第四位,就是这两个数组的中位数。
当两个数组的个数和为偶数时:
nums1:1 2 4 6
nums2:3 5 7 8
temp:1 2 3 4 5
temp主需要存储前5位,然后就可以得知中位数位4.5。
仅以此作为本人的学习记录。