C/C++教程

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

本文主要是介绍【每日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/

这篇关于【每日Leetcode-第四天】寻找两个正序数组的中位数的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!