HOT68-寻找两个正序数组的中位数

       leetcode原题链接：寻找两个正序数组的中位数

       上一篇：HOT67-寻找旋转排序数组中的最小值

       下一篇：HOT69-有效的括号

题目描述

       给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。算法的时间复杂度应该为 O(log (m+n)) 。

示例 1：

输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2

示例 2：

输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5

提示：

• nums1.length == m
• nums2.length == n
• 0 <= m <= 1000
• 0 <= n <= 1000
• 1 <= m + n <= 2000
• -106 <= nums1[i], nums2[i] <= 106

解题方法：

方法1: 采用类似于归并排序的归并法，时间复杂度为o(m+n)，空间复杂度为o(m+n)，每次排除1个元素。

方法2: 二分查找。比较两个元素nums1[i+k/2-1], nums2[j+k/2-1]的大小，如下情况

nums1： [i, i+1, i+2,... i+k/2-1] 共 k/2-1个元素

nums2：[j, j+1, j+2, ... j+k/2-1]共 k/2-1个元素

这两个数组中，小于等于min(nums1[i+k/2-1], nums2[j+k/2-1])的元素最少元素个数为i+k/2-1, 或者j+k/2-1，最多元素个数为(k/2-1+k/2-1)=k-2个，加上min(nums1[i+k/2-1], nums2[j+k/2-1]),最多k-1个，所以取k/2-1作为比较值是合理的。每次比较大小后，我们可以排除min(nums1[i+k/2-1], nums2[j+k/2-1])中对应的最小元素左边的值，每次排除k/2个元素。

C++代码

#include <iostream>
#include <vector>
/*
    nums1[k / 2 - 1]  k - 1 /2
    nums2[k / 2 - 1]
*/
class Solution {
public:
    double findMedianSortedArrays(std::vector<int>& nums1, std::vector<int>& nums2) {
        int n1 = nums1.size();
        int n2 = nums2.size();
        int r = (n1 + n2) % 2;
        if (r == 1) { //两个数组总个数为奇数
            int k = (n1 + n2) / 2 + 1; //k表示第k个
            return find_kth_num_v2(nums1, nums2, k);
        } 
        //两个数组总个数为偶数
        int k1 = (n1 + n2) / 2;
        int k2 = k1 + 1;
        return 0.5 * (find_kth_num_v2(nums1, nums2, k1) + find_kth_num_v2(nums1, nums2, k2));
    }
    
    //方法1: 归并思想(相当于每次淘汰1个数字)
    int find_kth_num_v1(std::vector<int>& nums1, std::vector<int>& nums2, int k) {
        int n1 = nums1.size();
        int n2 = nums2.size();
        if (n1 + n2 < k) {
            return -1;
        }
        int i = 0;//nums1的索引
        int j = 0;//nums2的索引
        while (i < n1 && j < n2) {
            if (nums1[i] <= nums2[j]) {
                if (k == 1) {
                    return nums1[i];
                }
                i++;
                k--;
            } else { // nums2[i] > nums2[j]
                if (k == 1) {
                    return nums2[j];
                }
                j++;
                k--;
            }
        }
        return i < n1 ? nums1[i + k - 1] : nums2[j + k - 1];//注意，这里k指的是还剩第k个
    }

    //方法2: 二分查找
    int find_kth_num_v2(std::vector<int>& nums1, std::vector<int>& nums2, int k) {
        int n1 = nums1.size();
        int n2 = nums2.size();
        if (n1 + n2 < k) {
            return -1;
        }
        int i = 0;//nums1的索引
        int j = 0;//nums2的索引
        while (1) {
            if (i == n1) {
                return nums2[j + k - 1];
            }
            if (j == n2) {
                return nums1[i + k - 1];
            }
            if (k == 1) {
                return std::min(nums1[i], nums2[j]);
            }
            int new_i = std::min(i + k / 2 - 1, n1 - 1);
            int new_j = std::min(j + k / 2 - 1, n2 - 1);
            if (nums1[new_i] <= nums2[new_j]) {
                k -= (new_i - i + 1); //注意：k和i的更新顺序不能换,先更新k
                i = new_i + 1; //后更新i
            } else {
                k -= (new_j - j + 1);//先更新k
                j = new_j + 1;//后更新j    
            }
        }
        return -1;
    }
};