leetcode 4. 寻找两个正序数组的中位数

leetcode 4. 寻找两个正序数组的中位数

题目详情

题目链接
给定两个大小为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。
请你找出这两个正序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。

• 示例 1:
  nums1 = [1, 3]
  nums2 = [2]
  则中位数是 2.0
• 示例 2:
  nums1 = [1, 2]
  nums2 = [3, 4]
  则中位数是 (2 + 3)/2 = 2.5

我的代码

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int len1 = nums1.size(), len2 = nums2.size();
        int len = len1 + len2;
        int loc1, loc2;
        if (len % 2) {
            loc1 = len / 2 + 1;
            loc2 = loc1;
        } else {
            loc1 = len / 2;
            loc2 = loc1 + 1;
        }
        vector<int> nums;
        int i = 0, j = 0, count = 0;
        while ((i < len1) && (j < len2)) {
            if (nums1[i] < nums2[j]) {
                nums.emplace_back(nums1[i++]);
            } else {
                nums.emplace_back(nums2[j++]);
            }
            if (++count == loc2) {
                return (nums[loc1 - 1] + nums[loc2 - 1]) / 2.0;
            }
        }
        if (i == len1) {
            while (j < len2) {
                nums.emplace_back(nums2[j++]);
                if (++count == loc2) {
                    return (nums[loc1 - 1] + nums[loc2 - 1]) / 2.0;
                }
            }
        } else {
            while (i < len1) {
                nums.emplace_back(nums1[i++]);
                if (++count == loc2) {
                    return (nums[loc1 - 1] + nums[loc2 - 1]) / 2.0;
                }
            }
        }
        return (nums[loc1 - 1] + nums[loc2 - 1]) / 2.0;
    }
};

我的成绩

执行结果：通过
执行用时 : 16 ms, 在所有 C++ 提交中击败了90.58%的用户
内存消耗 : 7.9 MB, 在所有 C++ 提交中击败了100.00%的用户

一些想法

我的方法是先找到中位数所在的位置，然后遍历两个数组来寻找，相当于暴力法。

执行用时为 0 ms 的范例

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m = nums1.size();
        int n = nums2.size();
        if(m > n){
            return findMedianSortedArrays(nums2, nums1);
        }
        int halflen = (m + n + 1) / 2;
        int mn = 0, mx = m;
        int i = (mn + mx) >> 1, j = halflen - i;
        double maxleft, minright;
        while(mn <= mx){
            i = (mn + mx) >> 1;
            j = halflen - i;
            if(i < mx && nums2[j-1] > nums1[i]){
                mn = i + 1;
            } else if(i > mn && nums1[i-1] > nums2[j]){
                mx = i - 1;
            } else {
                // 分别求出maxleft和minright
                if(i == 0){
                    // left part中不存在A
                    maxleft = nums2[j-1];
                } else if (j == 0){
                    maxleft = nums1[i-1];
                } else {
                    maxleft = max(nums1[i-1], nums2[j-1]);
                }
                if((m + n) % 2)
                    return maxleft;
                if(i == m){
                    minright = nums2[j];
                } else if (j == n){
                    minright = nums1[i];
                } else {
                    minright = min(nums1[i], nums2[j]);
                }
                return (minright + maxleft) / 2;
            }
        }
        return 0;
    }
};

思考

范例做法没看懂。。。