leetcode 4: Median of Two Sorted Arrays

最新推荐文章于 2023-06-12 16:30:30 发布

mikefan1991

最新推荐文章于 2023-06-12 16:30:30 发布

阅读量286

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/mikefan1991/article/details/48165077

leetcode 专栏收录该内容

201 篇文章

订阅专栏

本文介绍了一个使用归并排序思想解决寻找两个有序数组中位数问题的方法。通过比较两个数组元素，逐步确定中位数的位置，适用于奇数和偶数长度的数组组合。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Use the idea of merge sort.

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m=nums1.size();
        int n=nums2.size();
        int i=0,j=0,curr=1;
        double res1,res2;
        while(i<m&&j<n)
        {
            if(curr==(m+n)/2)
                res1=min(nums1[i],nums2[j]);
            if(curr==(m+n)/2+1)
            {
                res2=min(nums1[i],nums2[j]);
                break;
            }
            if(nums1[i]<=nums2[j])
                i++;
            else
                j++;
            curr++;
        }
        while(curr<=(m+n)/2+1)
        {
            if(curr==(m+n)/2)
            {
                if(i==m)
                    res1=nums2[j];
                else if(j==n)
                    res1=nums1[i];
            }
            if(curr==(m+n)/2+1)
            {
                if(i==m)
                    res2=nums2[j];
                else if(j==n)
                    res2=nums1[i];
            }
            if(i<m)
                i++;
            else if(j<n)
                j++;
            curr++;
        }
        if((m+n)%2)
            return res2;
        return (res1+res2)/2;
    }
};