leetcode 4.寻找两个有序数组的中位数（median of two sorted arrays）C语言

1.description

https://leetcode-cn.com/problems/median-of-two-sorted-arrays/description/

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

2.solution

参考leetcodecn 题解 的解法四

#define MAX(a, b) ((a) > (b) ? (a) : (b))
#define MIN(a, b) ((a) < (b) ? (a) : (b))

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){
    if(nums1Size>nums2Size){
        return findMedianSortedArrays(nums2, nums2Size, nums1, nums1Size);
    }

    int low = 0, high = nums1Size;
    
    while(low <= high){
        int i = low + (high - low) / 2;
        int j = (nums1Size + nums2Size + 1)/2 - i;
        if(i!=nums1Size && j!=0 && nums1[i]<nums2[j-1]){ // i小了
            low = i+1;
        }else if(i!=0 && j!= nums2Size && nums1[i-1]>nums2[j]){ //i大了
            high = i-1;
        }else{
            int max_l = 0;
            if(i==0){
                max_l = nums2[j-1];
            }else if(j==0){
                max_l = nums1[i-1];
            }else{
                max_l = MAX(nums1[i-1], nums2[j-1]);
            }
            
            // 总数为奇数个，返回MAX(nums1[i-1], nums2[j-1])
            if((nums1Size+nums2Size)%2!=0){
                return max_l;
            }

            int max_r = 0;
            if(i==nums1Size){
                max_r = nums2[j];
            }else if(j==nums2Size){
                max_r = nums1[i];
            }else{
                max_r = MIN(nums1[i], nums2[j]);
            }

            // 总数为偶数个，返回(MAX(nums1[i-1], nums2[j-1])+MIN(nums1[i], nums2[j]))/2.0
            return (max_l + max_r)/2.0;
        }
    }

    return 0.0;
}