LeetCode OJ : Median of Two Sorted Arrays

Median of Two Sorted Arrays 

链接：https://leetcode.com/problems/median-of-two-sorted-arrays/

题解：

给两个有序数组A,B，求中位数，要求时间复杂度为O(log (m+n))。

先求中位数的位置idx=(A.length+B.length)/2，分别比较两个数组的最小值哪个最小，弹出那个数，一直弹出idx次。如果全部数字为偶数，则中位数是第idx-1和idx次弹出的数；如果全部数字为奇数，则中位数是第idx次弹出的数。

代码：

public class Solution {
    public double findMedianSortedArrays(int A[], int B[]) {
        int lena=A.length;
        int lenb=B.length;
        if(lena+lenb==1){
            if(lena==0) return B[0];
            else        return A[0];
        }
        int mid=(lena+lenb)/2;
        int idxa=0;
        int idxb=0;
        int idx=0;
        double midValue=0.0,ans=0.0;
        for(;idx<=mid;++idx){
            ans=midValue;
            if(idxa==lena){
                midValue=B[idxb];
                ++idxb;
            }
            else if(idxb==lenb){
                midValue=A[idxa];
                ++idxa;
            }
            else if(A[idxa]<B[idxb]){
                midValue=A[idxa];
                ++idxa;
            }
            else{
                midValue=B[idxb];
                ++idxb;
            }
        }
        if((lena+lenb)%2==1){
            return midValue;
        }
        else{
            return (ans+midValue)/2;
        }
    }
}

来源： http://blog.youkuaiyun.com/acm_ted/article/details/44488325