【leetcode-数组】寻找两个正序数组的中位数

本文介绍了一种在O(log(m+n))时间复杂度内找到两个已排序数组中位数的方法。通过对比并合并两个数组的元素,仅需排序一半数量的元素即可找到中位数。

题目:

给定两个大小为 m 和 n 的正序(从小到大)数组 nums1 和 nums2

请你找出这两个正序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。

你可以假设 nums1 和 nums2 不会同时为空。

 

示例 1:

nums1 = [1, 3]
nums2 = [2]

则中位数是 2.0

示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

则中位数是 (2 + 3)/2 = 2.5

思路:

求中位数,即求两个数组合并后中间位置的数,但是题目已经给出两个数组是排好序的,所以只需将两个数组的一半之数排序即可。因为是两个已经排好序的数组,所以只需依次从头比较,较小的放入新建的数组之中即可。

java代码:

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int sum=nums1.length+nums2.length;
        int nums3[]=new int[sum];
        int i=0,j=0,temp=0;
        double res;
        while(temp<=sum/2)//将一半的数组排序即可
        {
            if(i<nums1.length&&j<nums2.length)//防止数组溢出
            {
                if(nums1[i]<nums2[j])
                {
                    nums3[temp++]=nums1[i++];
                }
                else
                {
                    nums3[temp++]=nums2[j++];
                }
            }
            else if(i>=nums1.length)//其中一个数组到末尾了
            {
                nums3[temp++]=nums2[j++];
            }
            else 
            {
                nums3[temp++]=nums1[i++];
            }
            
        }
        if(sum%2==0)//偶数位个
        {
            res=((double)(nums3[sum/2]+nums3[sum/2-1]))/2; //需要强转一下类型
        }
        else
        {
            res=nums3[sum/2];//奇数位个
            
        }
          
        return res;
     }
  }

 

由于水平有限,文章中难免会有一些错误,有纰漏之处恳请各位大佬不吝赐教!

及时更新最新文章和学习资料,一起来学习:

推荐阅读:

【leetcode-数组】买卖股票的最佳时机 II  - CSDN博客

 【leetcode-数组】存在重复

 【leetcode-数组】两数之和 II - 输入有序数组

【leetcode-数组】长度最小的子数组 

【leetcode-数组】 旋转数组 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值