二分5.寻找两个排序数组的中位数

首先如果使用归并数组找到中间数,则空间复杂度为O(n),时间复杂度为O(n),如果利用一个指针来达到找中位数的过程,模拟归并但不归并,那时间复杂度仍为O(n)
由于题中说了是有序的,所以可以尝试二分,这个题本质就是找到数组中第k小的数,k为(m+n)/2或者是(m+n)/2和(m+n)/2+1的平均值。
先比较A[k/2-1]和B[k/2-1],若A[k/2-1]<B[k/2-1],则比A[k/2-1]小的数肯定不可能成为k,则直接排除掉A[k/2-1]以及比它小的,可以排除掉k/2的数字,并根据排除掉的数字来减小k的值直到k为0.排除就利用index的增大来表现一个数组的开始在哪里,因为不会排除后面的,所以不用设置后一个。
如果k=1,则返回两个数组的首元素的最小值,如果一个数组为空,则直接可以返回另一个数组中第k个数字,如果越界,则选择最后一个元素,记得根据实际排除的值来减小k。

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int length1=nums1.length;
        int length2=nums2.length;
        int length=length1+length2;
        if((length&1) ==1){
            int k=length/2+1;
            return core(nums1,nums2,k);
        }else{
            int k1=length/2;
            int k2=length/2+1;
            return (core(nums1,nums2,k1)+core(nums1,nums2,k2))/2;
        }
    }
    public double core(int[] nums1,int[] nums2,int k){
        int i1=0;
        int i2=0;
        while(true){
            if(i1==nums1.length){
                return nums2[i2+k-1];
            }
            if(i2==nums2.length){
                return nums1[i1+k-1];
            }
             if(k==1){
                return  (double)Math.min(nums1[i1],nums2[i2]);
            }
            int half=k/2;
            int newIndex1=Math.min(half+i1,nums1.length)-1;
            int newIndex2=Math.min(half+i2,nums2.length)-1;
            if(nums1[newIndex1]<=nums2[newIndex2]){
                k-=(newIndex1-i1+1);
                i1=newIndex1+1;
            }else{
                k-=(newIndex2-i2+1);
                i2=newIndex2+1;
            }
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值