leetcode 4:兩個排序陣列的中位數
阿新 • • 發佈:2018-12-15
尋找中位數,當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; }