LeetCode 4 - Median of Two Sorted Arrays

原题如下：

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)).

Example 1:

nums1 = [1, 3]
nums2 = [2]

The median is 2.0

Example 2:

nums1 = [1, 2]
nums2 = [3, 4]

The median is (2 + 3)/2 = 2.5

简单方法：可以将两个有序数组做归并排序，然后找到中间元素并计算结果。其时间复杂度为O(n)，空间复杂度为O(n)。

更好一点的办法，是使用二分法。由于数组是有序的，可以利用这一特性，二分搜索找到这样的数，将两个数组平均的分成两半。通常情况下，算法复杂度为O(logn)，空间复杂度为O(1)。具体算法见如下代码：

    private int findNthSortedArrays(int A[], int as, int B[], int bs, int n){
        while(n>1 && as < A.length && bs < B.length){
            int half = n/2;
            if(as + half > A.length){
                half = A.length - as;
            }
            if(bs + half > B.length){
                half = B.length - bs;
            }
            if(A[as+half-1] > B[bs+half-1]){
                bs = bs + half;
            }else{
                as = as + half;
            }
            n = n - half;
        }
        if(as >= A.length){
            return B[bs + n-1];
        }else if(bs >= B.length){
            return A[as+n-1];
        }
        return A[as]>B[bs]?B[bs]:A[as];
    }


    public double findMedianSortedArrays(int A[], int B[]) {
        int len = A.length+B.length;
        if(len%2 == 0){
            int m1 = findNthSortedArrays(A, 0, B, 0, len/2);
            int m2 = findNthSortedArrays(A, 0, B, 0, len/2+1);
            return ((double)m1+m2)/2;
        }else{
            return findNthSortedArrays(A, 0, B, 0, len/2+1);
        }
    }

参考源码：https://github.com/pkufork/Martians/blob/master/src/main/java/com/pkufork/martians/leetcode/L4_MedianofTwoSortedArrays.java