LeetCode 4. Median of Two Sorted Arrays 寻找两个正序数组的中位数 Java 极简

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

Median of Two Sorted Arrays
Hard | 2021-1-1 @threepotatox

寻找两个正序数组的中位数
困难 | 2021-1-1 @有三只土豆

题目大意：

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

示例 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
• -10⁶ <= nums1[i], nums2[i] <= 10⁶

设计思路：

因为是两个有序数组，所以我们可以从小到大遍历两个数组中的元素，当计数到达总数的一半时，即找到中位值。

关键词：中位数

Java 版：

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int median1 = 0, median2 = 0;
        int len1 = nums1.length, len2 = nums2.length;
        int len = len1 + len2;
        int f, i = 0, j = 0;

        f = len % 2 == 1 ? 1 : 0;
        len = len / 2 + 1;
        while (len > 0) {
            median1 = median2;
            if (i < len1 && j < len2) {
                if (nums1[i] < nums2[j]) {
                    median2 = nums1[i];
                    i++;
                } else {
                    median2 = nums2[j];
                    j++;
                }
            } else {
                if (i < nums1.length) {
                    median2 = nums1[i];
                    i++;
                } else {
                    median2 = nums2[j];
                    j++;
                }
            }
            len--;
        }

        return f == 1 ? (double)median2 : (double)(median1 + median2) / 2.0;
    }
}