寻找两个正序数组的中位数(Median of Two Sorted Arrays)

题目

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

 

示例 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
示例 3：

输入：nums1 = [0,0], nums2 = [0,0]
输出：0.00000
示例 4：

输入：nums1 = [], nums2 = [1]
输出：1.00000
示例 5：

输入：nums1 = [2], nums2 = []
输出：2.00000
 

提示：

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.com/problems/median-of-two-sorted-arrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

一、合并数组

既然是求中位数，那么首先想到的就是将两个数组按照升序进行合并，再根据奇数和偶数分别返回结果。

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int l1 = nums1.length;
        int l2 = nums2.length;
        int length = l1 + l2;
        int[] array = new int[l1 + l2];
        int i = 0; 
        int j = 0;
        int index = 0;
        double res = 0;

        while(i < l1 || j < l2) {
           int num1 = i < l1 ? nums1[i] : Integer.MAX_VALUE; //数组元素不存在时，可以直接用最大数表示。这样不会影响正常排序
           int num2 = j < l2 ? nums2[j] : Integer.MAX_VALUE;
            if (num1 >= num2) {
                array[index++] = num2;
                j++;
            } else {
                array[index++] = num1;
                i++;
            }
        }

        if (length % 2 == 0) {
            res = (array[length / 2 - 1] + array[length / 2]) / 2.0;
        } else {
            res = array[length / 2];
        }

        return res;
    }
}

二、只求中位数，不需要生成新的数组

我们完全不需要生成一个数组，而是找到中位数即可。
但是奇数和偶数所返回的位数不一样，我们通过第一种解法，发现无论是奇数还是偶数，最后一位都是 总长度/2,只是偶数多了总长度/2 - 1.那么我们可以每次保存本次的数值和上一次的数值，再通过长度判断，决定返回什么数值。这两种情况的循环次数都为 总长度/2 + 1次。

public class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int l1 = nums1.length;
        int l2 = nums2.length;
        int length = l1 + l2;
        int i = 0;
        int j = 0;
        int now = 0;
        int last = 0;

        for (int index = 0; index <= length / 2; index++) {
            int num1 = i < l1 ? nums1[i] : Integer.MAX_VALUE;
            int num2 = j < l2 ? nums2[j] : Integer.MAX_VALUE;
            last = now;

            if (num1 >= num2) {
                now = nums2[j++];
            } else {
                now = nums1[i++];
            }
        }

        if (length % 2 == 0) {
            return (now + last) / 2.0;
        } else {
            return  now * 1.0;
        }
    }
}