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

最新推荐文章于 2024-10-26 19:20:21 发布

原创最新推荐文章于 2024-10-26 19:20:21 发布 · 90 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

273 篇文章

订阅专栏

本文介绍了一个C++程序，该程序用于合并两个已排序的整数数组，并找到合并后数组的中位数。首先，将两个数组合并到一个新的数组中，然后对这个数组进行排序。接着，根据数组的大小，程序会返回中位数，如果数组长度为奇数，则直接返回中间元素；如果为偶数，则返回中间两个元素的平均值。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        vector<int>obj;
        obj.insert(obj.end(),nums1.begin(),nums1.end());
        obj.insert(obj.end(),nums2.begin(),nums2.end());
        sort(obj.begin(),obj.end());

        for(int i=0;i<obj.size();i++)
        {
            cout<<obj[i]<<" ";
        }

        if(obj.size()==1)
        {
            return obj[0];
        }
        if(obj.size()==2)
        {
            return 0.5*(obj[0]+obj[1]);
        }
        if(obj.size()%2==1)
        {
            return obj[obj.size()/2];
        }
        cout<<obj[obj.size()/2]<<" "<<obj[obj.size()/2-1]<<endl;
        return ((double)obj[obj.size()/(double)2]+obj[obj.size()/2-1])/(double)2;
    }
};