leetcode 4 Median of Two Sorted Arrays c语言实现

leetcode 4 Median of Two Sorted Arrays  c语言实现

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

思路：对于一个长度为n的已排序数列a，

若n为奇数，中位数为a[n / 2 ] ,

若n为偶数，则中位数(a[n / 2 - 1] + a[n / 2 ]) / 2 

如果我们可以在两个数列A,B，求出第K=(m+n)/2小的元素，便可以解决该问题 不妨设数列A元素个数为n，数列B元素个数为m，各自升序排序，求第k小元素 取A[k / 2 - 1] B[k / 2 -1] 比较，分三种情况(AB总数分奇偶)：

如果 A[k / 2 - 1] < B[k / 2 - 1] 那么，A[0~k/2 -1]必然是在AB合并后的前K小元素中，可以将其抛弃

如果 A[k / 2 - 1] > B[k / 2 - 1] ，B[0~k/2 -1]必然在AB合并的前K个元素中，将B[0~k/2-1]舍弃，

如果相等的化，返回其中任意一个

于是得到了数据规模变小的同类问题，递归解决 如果 k / 2 大于某数列个数，所求元素必然不在另一数列的前k / 2个元素中，同上操作就好。

#define MIN(x, y)  x > y?y : x

double findK(int a[], int m, int b[], int n, int k) {
    int parta = MIN(k / 2, m);
    int partb = k - parta;
    if(m > n)
        return findK(b, n, a, m, k); //始终保证a数组的长度小于b数组的长度再进行后续的操作
    if(m == 0)
        return b[k - 1];
    if(k == 1)
        return MIN(a[0], b[0]);
    // 说明a[0~parta]这些数一定在合并ab两数组的前k个中，可以将这部分0~parta抹除
    if(a[parta - 1] < b[partb - 1]) 
        return findK(a + parta, m - parta, b, n, k - parta);
    else if(a[parta - 1] > b[partb - 1])
        return findK(a, m, b + partb, n - partb, k - partb);
    else
        return a[parta - 1]; //或者return b[partb - 1]
    
}

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
    int total = nums1Size + nums2Size;
    if(total & 0x1) //奇数
        return findK(nums1, nums1Size, nums2, nums2Size, total/2 + 1);
    else  //偶数
        return 0.5 * (findK(nums1, nums1Size, nums2, nums2Size, total/2) +
                      findK(nums1, nums1Size, nums2, nums2Size, total/2 + 1));  
}