算法训练:寻找两个有序数组的中位数

题目描述

给定两个大小为 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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays/
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

分析

若不考虑时间复杂度,我们可以将两个数组合并后进行排序,找第(m+n+1)/2位和第(m+n+2)/2位的元素,两者的平均数即为两个有序数组的中位数,然而,合并数组进行排序的时间复杂度为O(m+n),不符合题目要求。
此时,时间复杂度的要求为对数阶,因此我们想到用二分法,所谓的中位数是可以将一个有序集合分成两个相等长度的子集,题中规定的两个数组均为有序的,因此,我们需要在两个数组的左右两侧分别选出(m+n+1)/2个元素,并求出左子集的最大值与右子集的最小值的平均数,即为中位数。

代码

public class Solution {
public double FindMedianSortedArrays(int[] nums1, int[] nums2) {
        int m = nums1.Length;
        int n = nums2.Length;
        if (m > n)
        {
            int[] change = nums1;
            nums1=nums2;
            nums2=change;
            m=nums1.Length;
            n=nums2.Length;
        }
        int k = (m + n + 1)/2;
        int left = 0;
        int right = m;
        while (left < right)
        {
            int i = (left + right)/2;
            int j = k - i;
            if (nums1[i] < nums2[j - 1])
                left = i + 1;
            else
                right = i;
        }
        int m1 = left;
        int m2 = k - left;
        int a=0,b=0;
        if(m1!=0)
            a=nums1[m1-1];
        if(m2!=0)
            b=nums2[m2-1];
        int max = Math.Max(a,b);
        if ((m + n)%2 == 1)
            return max;
        int a1=0,b1=0;
        if(m1!=m)
            a1=nums1[m1];
        if(m2!=n)
            b1=nums2[m2];
        int min = Math.Min(a1,b1);
        return (max + min)*0.5;        
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值