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

题目：

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

思路一：

暴力解法，先将这两个数组进行合并，然后对合并了的数组进行排序；

最后通过下面这个式子

((m+n)%2==0)?(num[(m+n)/2]+num[(m+n)/2-1])/2.0:(double)num[(m+n)/2];

进行计算中位数。详细代码如下：

import java.util.*;
class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
           int m = nums1.length;int n = nums2.length;
           int[] num = new int[m + n];
           int i = 0;int j = 0;
           for(i = 0;i<m;i++) {
               num[i] = nums1[i];
           }
           for(j = 0;j < n;j++) {
               num[i++] = nums2[j];
           }
           Arrays.sort(num);
           return ((m+n)%2==0)?(num[(m+n)/2]+num[(m+n)/2-1])/2.0:(double)num[(m+n)/2];
    }
}

 

思路二：

可以使用一个类似于双指针的方法进行计算；首先，定义一个  i  是用来遍历数组nums1的，定义一个  j  是用来遍历数组nums2的。题目要求计算中位数，可以分为两种情况，一种为两个数组总和为奇数个的时候，中位数是 下标（m + n）/ 2对应的数；第二种情况是两个数组总和为偶数个的时候，中位数是 下标（m + n）/ 2 和 下标（m + n）/ 2-1对应的数字之和除以 2；所以需要循环（m + n）/ 2 次去寻找中位数，同时记录下第（m + n）/ 2个数字和第（m + n）/ 2 -1 个数字；用num1 去记录每一次被跳过的那个值，用num2 记录第（m + n）/ 2 -1 个数字，用num3记录第（m + n）/ 2个数字。会出现一个数组被遍历完然而还没有找到中位数的情况；详细代码如下：（切记if条件语句中的判断条件不能错位，同时else语句不能省略，会出现下标越界的情况）

import java.util.*;
class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int count = 0;
        int m = nums1.length;int n = nums2.length;
        int i = 0;int j = 0;
        int num1 = 0;int num2 = 0;int num3 = 0;
        for(count = 0;count <= (m + n)/2;count++) {
            if(i < nums1.length && j < nums2.length && nums1[i] < nums2[j]){num1 = nums1[i++];} 
        else if(j < nums2.length && i < nums1.length && nums1[i] >= nums2[j]) num1 = nums2[j++];
        else if(i == nums1.length) {num1 = nums2[j++];}
        else if(j == nums2.length) {num1 = nums1[i++];}
        if(count == (m + n) / 2 -1) num2 = num1; 
        if(count == (m + n) / 2)    num3 = num1;
        }
        if((m + n) % 2 != 0) return num3;
        else return (num3 + num2 ) / 2.0;
    }
}