Median of Two Sorted Arrays (全数组扫描)

public class Solution {


    public double findMedianSortedArrays(int A[], int B[]) {
        // Start typing your Java solution below
        // DO NOT write main() function
        int alength = A.length;
        int blength = B.length;
        
        if (alength == 0 ) {
            if (blength % 2 == 1) {
                return B[blength/2];
            }
            else {
                return ( B[blength/2 - 1] + B[blength/2] ) / 2.0; 
            }
        }
        else if (blength == 0 ) {
            if (alength % 2 == 1) {
                return A[alength/2];
            }
            else {
                return ( A[alength/2 - 1] + A[alength/2] ) / 2.0; 
            }
        }
        
        int left  = 0;
        int right = alength-1;
        
        return findMedian(A, B, left, right, alength, blength);
        
    }
    
    public double findMedian(int A[], int B[], int l, int r, int nA, int nB) {
        if (l>r) {
            int left  = 0;
            int right = nB-1;
            return findMedian(B, A, left, right, nB, nA);
        }


        
        int i = (l+r)/2;
        int j = (nA+nB)/2-i-1;
        
        boolean lvalid = false;
        if (j <= nB - 1 ) {
            lvalid = (j < 0) || (A[i] >= B[j]);
        }
        else {
        lvalid = (j < 0);
        }
        
        boolean rvalid = false;
        if (j+1 >=0 ) {
            rvalid = (j >= nB - 1) || (A[i] <= B[j + 1]);
        }
        else {
        rvalid = (j >= nB - 1);
        }
        
        if ( lvalid && !rvalid ) return findMedian(A, B, l, i-1, nA, nB);
        else if (!lvalid && rvalid) return findMedian(A, B, i+1, r, nA, nB);
        else {
            if ( (nA+nB)%2 == 1 ) return A[i];
            else if (i>0) {
                if (j<0) {
            return (A[i]+A[i-1])/2.0;
            }
            else {
            return (A[i]+Math.max(B[j], A[i-1]))/2.0;
            }
           
            }
            else return (A[i]+B[j])/2.0;
        }
    }


}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值