leetcode 寻找两个正序数组的中位数

leetcode 寻找两个正序数组的中位数_第1张图片
利用递归思想
两个数组,分别找到k/2的位置
将较小的那一组前k/2砍去,k减去砍去的部分
依次递归,直到k==1时结束递归;

class Solution {
     
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
     
        int tot=nums1.size()+nums2.size();
        if(tot%2==0){
     
            int left=find(nums1,0,nums2,0,tot/2);
            int right=find(nums1,0,nums2,0,tot/2+1);
            return (left+right)/2;
        }//总长为偶数,,,
        else{
     
            return find(nums1,0,nums2,0,tot);
        }//总长为奇数,,,
    }
    int find(vector<int>& nums1,int i,vector<int>& nums2,int j,int k){
     
        if(nums1.size()-i>nums2.size()-j) return find(nums2,j,nums1,i,k);//确保在第一个参数为较短的数组
        if(k==1){
     
            if(nums1.size()==i)
            return nums2[j];
            else return min(nums1[i],nums2[j]);
        }//递归出口
        if(nums1.size()==i) return nums2[j+k-1];//如果数组1已经遍历完了,就在数组二里找第k个数
        int si=min((int)nums1.size(),i+k/2),sj=j+k-k/2;
        if (nums1[si-1]>nums2[sj-1]) return find(nums1,i,nums2,sj,k-(sj-j));//砍掉小的一组sj-j=k-k/2;
        else return find(nums1,si,nums2,j,k-(si-i));//砍掉小的一组si-i=k/2;
    }
};

你可能感兴趣的:(C++数据结构)