【每日Leetcode-第四天】寻找两个正序数组的中位数

题目描述

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

解题思路

将两个有序数组归并为一个有序数组,直接获取对应下标的元素即可。

解题代码

// 归并两个数组
public int[]  merge(int[] nums1,  int[] nums2)  {
    int[] total  =  new int[nums1.length + nums2.length];
    // i: nums1的索引,j: nums2的索引, k: total的索引
    int i = 0, j = 0, k = 0;

    // 遍历并比较两个数组的元素并且将较小的元素放入total数组中
    while  (i < nums1.length  && j < nums2.length) {
        // 比较大小
        if (nums1[i] < nums2[j]) {
            total[k++] = nums1[i++];
        } else {
            total[k++] = nums2[j++];
        }
    }

    // 如果nums1数组还有元素
    while (i < nums1.length) {
        total[k++] = nums1[i++];
    }

    // 如果nums2数组还有元素
    while (j < nums2.length) {
        total[k++] = nums2[j++];
    }

    return total;
}


public double findMedianSortedArrays(int[] nums1, int[] nums2) {
    int[] total = merge(nums1, nums2);
    int len = total.length;

    if (total.length % 2 == 0) {
        return (total[len / 2 - 1] + total[len / 2]) / 2.0;
    } else {
        return total[len / 2] * 1.0;
    }
}

题目来源

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays/

上一篇:349. 两个数组的交集


下一篇:1818. 绝对差值和 力扣(中等) 二分+排序 lower_bound(vector.begin(),..)