leetcode 26: Median of Two Sorted Arrays


There are two sorted arrays A and B 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)).

<uncompleted, do not pass all test cases.>

class Solution {
private:
    int len;
    int m;
    int n;
public:
    double findMedianSortedArrays(int A[], int m, int B[], int n) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        Solution::m = m;
        Solution::n = n;
        len = m + n;
        
        if(true) {
            return medianRec( A, B,  0, m-1 ) ;
        } else {
            return medianRec( B, A, 0, n-1);
        }
    }
    
    double medianRec( int A[], int B[], int left, int right) {
        if( left > right) 
            return medianRec(B, A, max(0, len/2-m), min(n-1, len/2) ); 
        
        int i = (left+right) / 2;
        int j = len/2 - i - 1;
        
        if (j>=0 && A[i] < B[j]) return medianRec(A, B, i+1, right);
        else if (j<n-1 && A[i] > B[j+1]) return medianRec(A, B, left, i-1);
        else {
            if ( len % 2 == 1 ) return A[i];
            else if (i>0) return (A[i]+max(B[j], A[i-1]))/2.0;
            else return (A[i]+B[j])/2.0;
         }
    }
};


posted @ 2013-01-10 08:18  西施豆腐渣  阅读(113)  评论(0编辑  收藏  举报