寻找两个正序数组的中位数(困难)

题目链接
给定两个大小为 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

题解:https://leetcode-cn.com/problems/median-of-two-sorted-arrays/solution/xiang-xi-tong-su-de-si-lu-fen-xi-duo-jie-fa-by-w-2/

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int m=nums1.length;
        int n=nums2.length;
        if(m>n){
            return findMedianSortedArrays(nums2,nums1);
        }
        if(m==0){
            if(n%2==1) return nums2[n/2];
            else return (nums2[n/2]+nums2[n/2-1])*0.5;
        }
        int l=0,r=m;
        int mx=0,mn=0;
        while(l<=r){
            int i=(l+r)>>1;
            int j=(m+n+1)/2-i;      
            if(i==0) mx=nums2[j-1];
            else if(j==0) mx=nums1[i-1];
            else mx=Math.max(nums1[i-1],nums2[j-1]);
            if(i==m) mn=nums2[j];
            else if(j==n) mn=nums1[i];
            else mn=Math.min(nums1[i],nums2[j]);
            if(mx<=mn) return (m+n)%2==1?mx:(mx+mn)*0.5;
            else{
                if(i==0) l=i+1;
                else if(j==0) r=i-1;
                else{
                    if(nums1[i-1]>=nums2[j-1]) r=i-1;
                    else l=i+1;
                }
            }
        }
        return 0.0;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值