LeetCode:Median of two sorted array

来源:互联网 发布:知乎清华五道口 编辑:程序博客网 时间:2024/05/29 13:20

double findkth(int *nums1,int nums1Size,int *nums2,int nums2Size,int k){    int  a,b;if(nums1Size<nums2Size){     a=(k/2<nums1Size)?k/2:nums1Size; b=k-a;}else{b=(k/2<nums2Size)?k/2:nums2Size;a=k-b;}    if(nums1Size==0)return nums2[k-1];    if(nums2Size==0)return nums1[k-1];    if(k==1)return (nums2[k-1]>nums1[k-1])?nums1[k-1]:nums2[k-1];        if(nums1[a-1]<nums2[b-1])return findkth(nums1+a,nums1Size-a,nums2,nums2Size,k-a);    else if(nums1[a-1]==nums2[b-1])return  nums1[a-1];    else return findkth(nums1,nums1Size,nums2+b,nums2Size-b,k-b);}double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {    int k=nums1Size+nums2Size;    if(k%2==0)//printf("%f %f",findkth(nums1,nums1Size,nums2,nums2Size,k/2), findkth(nums1,nums1Size,nums2,nums2Size,k/2+1));return (findkth(nums1,nums1Size,nums2,nums2Size,k/2)+findkth(nums1,nums1Size,nums2,nums2Size,k/2+1))/2;    else //printf("%f",findkth(nums1,nums1Size,nums2,nums2Size,k/2));return findkth(nums1,nums1Size,nums2,nums2Size,k/2+1);}

求两个排序数组的中位数:这是一个O(log(n+m))的算法。将求中位数转化为求两个数组中第(m+n)/2大的数,进一步扩展为求第k大的数。

首先考虑三种边界情况:两个数组中有一个数组长度为0,则返回另一个数组的中位数即可;若k=1,则只需输出两个数组各自最小值中较小的那一个。

然后,要求第k大的数,先假设两个数组各取前k/2,如果,第一个数组中第k/2个数字小于第二个数组中第k/2个数字,则说明第一个数组中前k/2个都小于整体第k个,故只需在后半部分找。反之同理。若 刚好相等,则该值为第k个。


0 0
原创粉丝点击