寻找两个有序数组的中位数 O(log(m+n))AND O(log(min(n,m)))

O(log(m+n)) :

不断排除某个数组的前缀而相应的更改K，那么当K=1或其中一个数组长度为0时结束。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int n=nums1.size(),m=nums2.size();
        return 1.*(getk(nums1,0,n-1,nums2,0,m-1,(n+m+1)/2)+getk(nums1,0,n-1,nums2,0,m-1,(n+m+2)/2))/2;
    }
    int getk(vector<int>& nums1,int l1,int r1,vector<int>& nums2,int l2,int r2,int k) {
        int len1=r1-l1+1;
        int len2=r2-l2+1;
        if(len1>len2) return getk(nums2,l2,r2,nums1,l1,r1,k);
        if(len1==0) return nums2[l2+k-1];
        if(k==1) return min(nums1[l1],nums2[l2]);
        int x=l1+min(len1,k/2)-1;
        int y=l2+min(len2,k/2)-1;
        if(nums1[x]<nums2[y]) {
            return getk(nums1,x+1,r1,nums2,l2,r2,k-(x-l1+1));
        }
        else {
            return getk(nums1,l1,r1,nums2,y+1,r2,k-(y-l2+1));
        }
    }
};

O(log(min(n,m))

利用中位数的划分思想使问题变成找到最大的i满足,max(a[i-1],b[j-1])<=min(a[i],b[j])其中j=(n+m+1)/2-i；
还有挺多小细节需要注意

class Solution {
    const int inf = 0x3f3f3f3f;
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int n=nums1.size(),m=nums2.size();
        if(n>m) {
            swap(n,m);
            swap(nums1,nums2);
        }
        int l=0,r=n;
        int m1=0,m2=0;
        while(l<=r) {
            int i=(l+r)>>1;
            int j=((n+m+1)>>1)-i;
            int a=i?nums1[i-1]:-inf;
            int b=i==n?inf:nums1[i];
            int c=j?nums2[j-1]:-inf;
            int d=j==m?inf:nums2[j];
            if(a<=d) {
                m1=max(a,c);
                m2=min(b,d);
                l=i+1;
            }
            else r=i-1;
        }
        return ((n+m)&1^1)?1.*(m1+m2)/2:m1;

    }
    
};