LeetCode4-寻找两个正序数组的中位数

50070

于 2022-04-22 13:58:43 发布

阅读量64

点赞数

分类专栏： LeetCode 文章标签：算法

本文链接：https://blog.csdn.net/l15570068307/article/details/124344016

版权

LeetCode 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

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

2、解法分析

数组为正序数组，将两个数组合并即可求得中位数。具体逻辑如下：
1、循环m+n 次对其赋值，把两个数组合并为一个数组
2、如果m+n 为偶数，则以示例二方式计算中位数
3、如果m+n 为奇数，则以示例一方式计算中位数


    public double findMedianSortedArrays(int[] nums1, int[] nums2) {

        //假定第一位为中位数
        int m = 0;  //代表num1数组下标
        int n = 0;   //代表num2数组下标
        int length = nums1.length + nums2.length;
        int[] temp = new int[length];

        for (int i = 0; i < length; i++) {
            if (m < nums1.length && n < nums2.length) {
                if (nums1[m] < nums2[n]) {
                    temp[i] = nums1[m];
                    m++;
                    continue;
                } else {
                    temp[i] = nums2[n];
                    n++;
                    continue;

                }
            }

            if (m >= nums1.length) {
                temp[i] = nums2[n];
                n++;
                continue;
            }

            if (n >= nums2.length) {
                temp[i] = nums1[m];
                m++;
                continue;
            }
        }
        if (length % 2 == 0) {
            return (temp[length / 2] + temp[length / 2 - 1]) / 2f;  //切记，分母不能是整数，否则会被转为整型，小数位会被舍去。
        } else {
            return (temp[(length - 1) / 2]);
        }
    }