寻找两个正序数组的中位数 王道解法

题目描述

 

给定两个大小分别为 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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/median-of-two-sorted-arrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

题解分析

1、看到时间复杂度为 O(log (m+n))，说明遍历一次两个数组就能得出中位数，但是题目对空间复杂度不做要求，典型的用空间换时间的做法。

2、看到是有序的，可以使用归并算法。

思路：用空间换时间，开辟一个一维数组，类似归并排序的思想，用两个指针同时遍历两个数组，并将值从小到大复制到数组中。然后判断数组长度是否为偶数，若为偶数，则取中间两个数的均值，否则取中间值。因为数组是随机读取，所以在取中间值的时候直接用下标取值，不需要再遍历一次数组元素

代码c++

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        double res;
        int m=nums1.size();
        int n=nums2.size();
        int* a=new int [m+n];
        int index=0;
        int i,j;
        for(i=0,j=0;i<m&&j<n;){
            if(nums1[i]<=nums2[j]){
                a[index++]=nums1[i++];
            }else{
                a[index++]=nums2[j++];
            }
        }
        while(i<m){
            a[index++]=nums1[i++];
        }
        while(j<n){
            a[index++]=nums2[j++];
        }
        return index%2==0?1.0*(a[index/2]+a[index/2-1])/2:a[index/2];
    }
};