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

最新推荐文章于 2024-04-18 15:04:31 发布

原创最新推荐文章于 2024-04-18 15:04:31 发布 · 117 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#困难

编程刷题同时被 2 个专栏收录

13 篇文章

订阅专栏

LeetCode

12 篇文章

订阅专栏

该博客探讨了如何在O(log(m+n))的时间复杂度内找到两个有序数组nums1和nums2的中位数。通过示例1和示例2说明了算法的应用，并详细解释了解题思路，包括处理数组为空或其中一个数组先遍历完的情况。

题目描述

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

解题思路

分别遍历两个有序数组，先访问比较小的元素，直到中位数的位置为止。需要注意的点包括：

两个数组如果有一个先遍历完的情况；
两个数组如果有一个为空的情况；

代码实现

#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){
    double ans;
    int temp1;
    int temp2;
    int numsSize;
    int i;
    int j;
    int k;
    
    numsSize = nums1Size + nums2Size;
    i = 0;
    j = 0;
    k = 0;

    if(nums1Size!=0&&nums2Size!=0){
        while(k<=numsSize/2){
            if(i==nums1Size){
                if(numsSize%2==0&&k==numsSize/2-1){
                    temp2 = nums2[j++];
                }else{
                    temp1 = nums2[j++];
                }
            }else if(j==nums2Size){
                if(numsSize%2==0&&k==numsSize/2-1){
                    temp2 = nums1[i++];
                }else{
                    temp1 = nums1[i++];
                }
            }else{
                if(numsSize%2==0&&k==numsSize/2-1){
                    if(nums1[i]<nums2[j]){
                        temp2 = nums1[i++];
                    }else{
                        temp2 = nums2[j++];
                    }
                }else{
                    if(nums1[i]<nums2[j]){
                        temp1 = nums1[i++];
                    }else{
                        temp1 = nums2[j++];
                    }
                }
            }
            k++;
        }
        
        if(numsSize%2==0){
            ans = (double)((temp1+temp2))/2;
        }else{
            ans = temp1;
        }
    }
    if(nums2Size==0){
        if(nums1Size%2==0){
            ans = (double)((nums1[nums1Size/2-1]+nums1[nums1Size/2]))/2;
        }else{
            ans = nums1[nums1Size/2];
        }
    }
    
    if(nums1Size==0){
        if(nums2Size%2==0){
            ans = (double)((nums2[nums2Size/2-1]+nums2[nums2Size/2]))/2;
        }else{
            ans = nums2[nums2Size/2];
        }
    }

    return ans;
}