LeetCode 4 Median of Two Sorted Arrays

最新推荐文章于 2025-09-12 17:31:31 发布

weixin_30345055

最新推荐文章于 2025-09-12 17:31:31 发布

阅读量47

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/dacc123/p/9934100.html

本文介绍了一种解决LeetCode1004题目的高效算法，通过将两个已排序的数组合并并找到中位数，避免了使用快速排序导致的时间复杂度过高问题。该算法的时间复杂度为O(m+n)，提供了一种简单且高效的解决方案。

LeetCode 1004

思路

一开始我用快速排序将两个数组重新排序，居然超时。
其实两个已经排好的数组用一个for循环排序就好了，效率O(m+n) ，而快排是O((m+n)*log(m+n))

但是题目上给的是O(log(m+n))的效率，应该是把O(m+n)都放过了。

class Solution {
public:
    int a[1000005];
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        
        int n=nums1.size();
        int m=nums2.size();
        int i=0,j=0;
        int tag=0;
        while(i<n&&j<m)
        {
           if(nums1[i]<=nums2[j])
                a[tag++]=nums1[i++];
           else 
               a[tag++]=nums2[j++];
        }
        while(i<n)
        {
            a[tag++]=nums1[i++];
        }
        while(j<m)
        {
            a[tag++]=nums2[j++];
        }
        
        if((n+m)%2==0)
            return 0.5*(a[(n+m)/2]+a[(n+m)/2-1]);
        else
            return 1.0*a[(n+m)/2];   
    }
    
    
};

转载于:https://www.cnblogs.com/dacc123/p/9934100.html