Median of Two Sorted Arrays

最新推荐文章于 2023-10-07 04:23:19 发布

顺其自然__

最新推荐文章于 2023-10-07 04:23:19 发布

阅读量391

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 题目

本文链接：https://blog.csdn.net/JSRGFJZ/article/details/50154647

Leetcode 题目专栏收录该内容

110 篇文章

订阅专栏

本文介绍了一种在O(log(m+n))的时间复杂度内找到两个已排序数组中位数的方法。通过递归地查找第k个元素，文章提供了一个C++实现的解决方案，并详细解释了其背后的逻辑。

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

题目：There are two sorted arrays nums1 and nums2 of size m and n respectively. Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

思路：

本题如此讲解，设两个数组：a[0...i....m] , b[0...j....n]，两数组排列完毕。

如果a[i]<b[j] ，那么对a数组从i开始迭代寻找，前面那段没必要寻找。

循环递归结束的条件是k=1的时候，返回第一个数字。

代码：

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int len=nums1.size()+nums2.size();
        
        if(len%2)//奇数
            return  findKthElement( nums1 ,nums2 , (len+1)/2 ) ;
        else//偶数    
            return  (findKthElement(nums1,nums2,len/2)+findKthElement( nums1 , nums2 , len/2+1 ))*0.5 ;
    }
    
    double min(double x,double y){
        return x<y?x:y;
    }
    
    double findKthElement(vector<int>& nums1,vector<int>& nums2,int k){
        if(nums1.size()>nums2.size()){
            return findKthElement(nums2,nums1,k);
        }
        
        if(nums1.empty())   return nums2[k-1];
        if(nums2.empty())   return nums1[k-1];
        
        if(k==1)    return min(nums1[0],nums2[0]);
        
        int ia=min(k/2,nums1.size());
        int ib=k-ia;
        if(nums1[ia-1]<nums2[ib-1]){
            vector<int> new_nums1(nums1.begin()+ia,nums1.end());
            return findKthElement( new_nums1 , nums2 , k-ia ) ;
        }if(nums1[ia-1]>nums2[ib-1]){
            vector<int> new_nums2(nums2.begin()+ib,nums2.end());
            return findKthElement( nums1 , new_nums2 , k-ib ) ;
        }
        return nums1[ia-1];
    }
};