leetcode 4: Median of Two Sorted Arrays

原创于 2015-09-01 22:20:55 发布 · 286 阅读

0 ·

CC 4.0 BY-SA版权

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;
    }
};