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
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
   // nums1Size = sizeof(nums1)/sizeof(int);
   // nums2Size = sizeof(nums2)/sizeof(int);
    int num[nums1Size+nums2Size];
    int t = 0;
    double mid = 0;
    for(int i=0; i<nums1Size; i++)
    {
        num[i] = nums1[i];
    }
    for(int i=nums1Size; i<nums1Size+nums2Size; i++)
    {
        num[i] = nums2[i-nums1Size];
    }
    for(int i=0; i<nums1Size+nums2Size; i++)
    {
        for(int j=i+1; j<nums1Size+nums2Size; j++)
            if(num[i] > num[j])
            {
                  t = num[j];
                num[j] = num[i];
                num[i] = t;
            }
              
    }
    if((nums1Size+nums2Size) != 0)
    {
        if((nums1Size+nums2Size)%2 == 0)
            mid = ((double)num[(nums1Size+nums2Size)/2-1] + (double)num[(nums1Size+nums2Size)/2])/2;
        else
            mid = (double)num[(nums1Size+nums2Size-1)/2];
    }
    else
            return 0;
    return mid;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章