寻找两个有序数组的中位数

LeetCode 4. 寻找两个有序数组的中位数

  给定两个大小为 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

int findkth(int *nums1, int s1, int e1, int *nums2, int s2, int e2, int k) {
    int mid1 = s1 + (e1 - s1) / 2;
    int mid2 = s2 + (e2 - s2) / 2;
    if(e1 < s1)
        return nums2[s2 + k - 1];
    if(e2 < s2)
        return nums1[s1 + k - 1];
    int len = mid1 + mid2 - s1 - s2 + 2;
    if(nums1[mid1] < nums2[mid2]) {
        if(len > k)
            return findkth(nums1, s1, e1, nums2, s2, mid2 - 1, k);
        else
            return findkth(nums1, mid1 + 1, e1, nums2, s2, e2, k - (mid1 - s1 + 1));
    }
    else {
        if(len > k)
            return findkth(nums1, s1, mid1 - 1, nums2, s2, e2, k);
        else
            return findkth(nums1, s1, e1, nums2, mid2 + 1, e2, k - (mid2 - s2 + 1));
    }
}
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){
    int sum = nums1Size + nums2Size;
    if(sum % 2)
        return (double)findkth(nums1, 0, nums1Size - 1, nums2, 0, nums2Size - 1, sum / 2 + 1);
    else {
        int x1 = findkth(nums1, 0, nums1Size - 1, nums2, 0, nums2Size - 1, sum / 2 + 1);
        int x2 = findkth(nums1, 0, nums1Size - 1, nums2, 0, nums2Size - 1, sum / 2);
        return (double)(x1 + x2) / 2;
    }
}

  现在就是很后悔，早知道先做了，期末考白白丢分……
  这个写法采用了递归的写法，二分查找第k大的数或者第k大的数以及第k+1大的数，若采用不递归的写法：
算法如下：

https://leetcode-cn.com/problems/median-of-two-sorted-arrays/solution/xun-zhao-liang-ge-you-xu-shu-zu-de-zhong-wei-shu-b/

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){
    int imin = 0, imax = nums1Size < nums2Size ? nums1Size : nums2Size;
    if(nums1Size > nums2Size) {
        int *tmp, m;
        tmp = nums2;
        nums2 = nums1;
        nums1 = tmp;
        m = nums1Size;
        nums1Size = nums2Size;
        nums2Size = m;
    }
    while(imin <= imax) {
        int i = (imin + imax) / 2;
        int j = (nums1Size + nums2Size + 1) / 2 - i;
        if(j > 0 && i < nums1Size && nums1[i] < nums2[j - 1])
            imin = i + 1;
        else if(j < nums2Size && i > 0 && nums1[i - 1] > nums2[j])
            imax = i - 1;
        else {
            int maxleft = 0;
            if(i == 0)
                maxleft = nums2[j - 1];
            else if(j == 0)
                maxleft = nums1[i - 1];
            else 
                maxleft = nums1[i - 1] > nums2[j - 1] ? nums1[i - 1] : nums2[j - 1];
            if((nums1Size + nums2Size) % 2)
                return maxleft;
            int minright = 0;
            if(i == nums1Size)
                minright = nums2[j];
            else if(j == nums2Size)
                minright = nums1[i];
            else
                minright = nums1[i] < nums2[j] ? nums1[i] : nums2[j];
            return (double)(maxleft + minright) / 2;
        }
    }
    return 0.0;
}