1. 程式人生 > >leetcode 4:兩個排序陣列的中位數

leetcode 4:兩個排序陣列的中位數

尋找中位數,當m+n是奇數時,中位數為第(m+n+1)/2個數,當m+n為偶數時,中位數為第(m+n+1)/2和(m+n+2)/2的平均數

根據整數的取整,我們可以統一為中位數為第(m+n+1)/2和(m+n+2)/2的平均數。

如何使用二分法求兩個有序陣列的第k個數

首先將陣列A和陣列B分為left_A(下標為0~k/2-1),right_A,left_B(下標為0~k/2-1),right_B

如果left[k/2-2]>right[k/2] 第k個值在left_A,right_A,right_B中

反之,第k個值在left_B,right_A,right_B中

int getKth(int* ln, int m, int* rn, int n, int k) {
    if (m > n) {
        return getKth(rn, n, ln, m, k);
    }
    if (m == 0) {
        return rn[k - 1];
    }
    if (k == 1) {
        return std::min(ln[0], rn[0]);
    }
    int i = std::min(m, k / 2), j = std::min(n, k / 2);
    if (ln[i - 1] > rn [j - 1]) {
        return getKth(ln, m, rn + j, n - j, k - j);
    } else {
        return getKth(ln + i, m - i, rn, n, k - i);
    }
}
double findMedianSortedArrays(std::vector<int>& nums1, std::vector<int>& nums2) {
    int m = nums1.size();
    int n = nums2.size();
    int l=(m+n+1)/2;
    int r=(m+n+2)/2;
    int *ln=m>0?&nums1[0]:NULL;
    int *rn=n>0?&nums2[0]:NULL;
    return (getKth(ln,m,rn,n,l)+getKth(ln,m,rn,n,r))/2.0;
}