There are two sorted arrays nums1 and nums2 of size m and n respectively.
Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).
Example 1:
nums1 = [1, 3]
nums2 = [2]
The median is 2.0
Example 2:
nums1 = [1, 2]
nums2 = [3, 4]
The median is (2 + 3)/2 = 2.5
思路:要求时间复杂度O(log(n+m)),所以考虑采用二分查找.
由于两个数组都是有序的,那么结合之后的中位数就是大数组中第K大的数.
这里要分奇偶讨论一下,不详细介绍.
假设两个数组 a 和 b.找到a 和 b 中第 k/2大的数 a[k/2-1]和 b[k/2-1];记为 t1 和t2.
1.若 t1 == t2,那么t1 就是要找的中位数;
2.若 t1 < t2,那么 a 数组的前 t1 个数绝对比中位数小(这里很容易证明),所以抛弃 a[0]~a[t1-1], b不变,进行下一轮的查找;
3.若 t1 > t2,那么类似 2 中的步骤.舍去 b 数组中的一些值.
4.注意边界条件的一些处理.
int get_kth(int a[],int b[],int lena,int lenb,int k)
{
if(lena > lenb)//保证第一个数组元素更少
return get_kth(b, a, lenb, lena, k);
if(lena == 0)
return b[k-1];
if(k == 1)
return min(a[0],b[0]);
int mida = min(k/2,lena), midb = k-mida;
if(a[mida-1] == b[midb-1])
return a[mida-1];
else if(a[mida-1] < b[midb-1])
return get_kth(a+mida, b, lena-mida, lenb, k-mida);
else
return get_kth(a, b+midb, lena, lenb-midb, k-midb);
}
int min(int a,int b)
{
return a<b?a:b;
}
double findMedianSortedArrays(int* nums1, int n, int* nums2, int m)
{
int sum = n+m;
if(sum%2 == 1)//奇数
return get_kth(nums1,nums2,n,m,sum/2+1);
else
return ((get_kth(nums1, nums2, n, m, sum/2))+(get_kth(nums1, nums2, n, m, sum/2+1)))/2.0;
}