65.Median of two Sorted Arrays

2017-07-12  本文已影响0人  博瑜
class Solution {
/**
 * @param A: An integer array.
 * @param B: An integer array.
 * @return: a double whose format is *.5 or *.0
 */
public double findMedianSortedArrays(int[] A, int[] B) {
    // write your code here
    int lenA = A.length;
    int lenB = B.length;
    int totalLen = lenA + lenB;
    if (totalLen % 2 == 0) {
        int sum = findKth(A, 0, B, 0, totalLen / 2) + findKth(A, 0, B, 0, 1 + totalLen / 2);
        return sum / 2.0;
    } else {
        return findKth(A, 0, B, 0, 1 + totalLen / 2);
    }
}

private int findKth(int[] A, int startA, int[] B, int startB, int k) {
    if (startA >= A.length) return B[startB + k - 1];
    if (startB >= B.length) return A[startA + k - 1];
    if (k == 1) return Math.min(A[startA], B[startB]);
    int a = ((startA + k / 2) > A.length) ? Integer.MAX_VALUE : A[startA + k / 2 - 1];
    int b = ((startB + k / 2) > B.length) ? Integer.MAX_VALUE : B[startB + k / 2 - 1];
    if (a == b) {
        return a;
    } else if (a > b) {
        return findKth(A, startA, B, startB + k / 2, k - k / 2);
    } else {
        return findKth(A, startA + k / 2, B, startB, k - k / 2);   
    }
}
}
上一篇下一篇

猜你喜欢

热点阅读