LeetCode - 4. 两个排序数组的中位数（C语言）

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

代码

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {

    int numSize = nums1Size+nums2Size;
    int s[numSize];
    int i,l=0,r=0;

    for (i = 0; i <= (numSize/2);) {

        if (l == nums1Size){
            while (i <= (numSize/2)){
                s[i++] = nums2[r++];
            }
        } else if (r == nums2Size){
            while (i <= (numSize/2)){
                s[i++] = nums1[l++];
            }
        } else{
            if(nums1[l] < nums2[r]){
                s[i++] = nums1[l++];
            } else{
                s[i++] = nums2[r++];
            }
        }
//        for (int j = 0; j < numSize; ++j) {
//            printf("%d-",s[j]);
//        }
//        printf("\n");

    }
//    for (int j = 0; j < numSize; ++j) {
//        printf("%d-",s[j]);
//    }
//
//    printf("\n");

    if(numSize%2==1){
        return s[i-1] ;
    }else{
        return ((s[i-2]+s[i-1]) / 2.0);
    }

}

#include <stdio.h>
#include "LeetCode4.cpp"

int main() {
    double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size);

    int a[4] = {1,2};
    int b[3] = {3,4};
    double num = findMedianSortedArrays(a,2,b,2);
    printf("%.1f",num);
    return 0;
}