LeetCode 4 Median of Two Sorted Arrays【经典题目】

传送门

题意: 给定两个有序数组, 要求在O(log(m+n))的复杂度之内求出合并后的中位数

思路: 数据水，线性的也能过. 但是面试遇到就不一定能过了. 所以还是得想正解. 有个log，所以我们用二分来做, 具体做法就是每次算一算第k/2在哪里, 分别取出中间的来比较, 再按照大小关系递归算出，第k大在哪里, 每次递归减少k/2, 所以就是logk(k ≈ m + n), 所以复杂度合理!

AC Code

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int len1 = nums1.size(), len2 = nums2.size();
        int l = (len1 + len2 + 1) / 2;
        int r = l;
        if ((len1 + len2) % 2 == 0) ++ r;
        return (1.0*getKnum(nums1, 0, nums2, 0, l) + getKnum(nums1, 0, nums2, 0, r)) / 2.0;
    }
    int getKnum(vector<int>& A, int sa, vector<int>& B, int sb, int k) {
        if (sa >= A.size()) return B[sb + k -1];
        if (sb >= B.size()) return A[sa + k -1];
        if (k == 1) return min(A[sa], B[sb]);
        
        int l = INT_MAX, r = INT_MAX;
        if (sa + k/2 - 1 < A.size()) l = A[sa+k/2-1];
        if (sb + k/2 - 1 < B.size()) r = B[sb+k/2-1];
        return l < r ? getKnum(A, sa+k/2, B, sb, k - k/2) : getKnum(A, sa, B, sb+k/2, k - k/2);
    }
};