力扣HOT100 - 4. 寻找两个正序数组的中位数

_OLi_

已于 2024-05-09 19:24:51 修改

阅读量374

点赞数 8

分类专栏：力扣HOT100 算法题合集文章标签： leetcode 算法 java

于 2024-05-09 19:24:35 首次发布

本文链接：https://blog.csdn.net/qq_61504864/article/details/138627855

版权

力扣HOT100 算法题合集专栏收录该内容

100 篇文章 0 订阅

订阅专栏

解题思路：

两个数组合并，然后根据奇偶返回中位数。

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int m = nums1.length;
        int n = nums2.length;
        int[] nums = new int[m + n];
        if (m == 0) {
            if (n % 2 == 0) return (nums2[n / 2 - 1] + nums2[n / 2]) / 2.0;
            else return nums2[n / 2];
        }
        if (n == 0) {
            if (m % 2 == 0) return (nums1[m / 2 - 1] + nums1[m / 2]) / 2.0;
            else return nums1[m / 2];
        }

        int cnt = 0;
        int i = 0, j = 0;
        while (cnt != (m + n)) {
            if (nums1[i] < nums2[j]) nums[cnt++] = nums1[i++];
            else nums[cnt++] = nums2[j++];

            if (i == m) {
                while (j != n)
                    nums[cnt++] = nums2[j++];
            }
            if (j == n) {
                while (i != m)
                    nums[cnt++] = nums1[i++];
            }
        }
        if (cnt % 2 == 0) return (nums[cnt / 2 - 1] + nums[cnt / 2]) / 2.0;
        else return nums[cnt / 2];
    }
}