每日一练:LeetCode4 寻找两个正序数组的中位数

题目:
在这里插入图片描述
题目是给我们两个有序数组,然后合并后求它的中位数。我们可以创建一个新的数组,来储存他们合并后的有序数组,再来求得新数组的中位数就可以了。如果长度是奇数,那中位数就是最中间那个数,如果是偶数,则是中间两个数和的一半。
代码:

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        double[] nums3=new double[nums1.length+nums2.length];
        int len3=nums3.length;
        int i=0,j=0,z=0;
        while(i<nums1.length&&j<nums2.length){
            if(nums1[i]<nums2[j]){
                nums3[z++]=nums1[i++];
            }else{
                nums3[z++]=nums2[j++];
            }
        }
        while(i<nums1.length){
            nums3[z++]=nums1[i++];
        }
        while(j<nums2.length){
            nums3[z++]=nums2[j++];
        }
        if(len3%2==1){
            return nums3[len3/2];
        }else{
            return (nums3[len3/2-1]+nums3[len3/2])/2;
        }
    }
}

但是,这个方法时间复杂度为O(m+n),但题目要求的复杂度为O(log (m+n)),所以我们要找其他方法。求中位数,相当于就是求第k小的数,我们就可以使用二分法来解决此题,每次排除比第k小的数小的数字,最后求得第k小的数字,也就是中位数。整体用的是递归的方法。
代码:

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int n=nums1.length;
        int m=nums2.length;
        // 找出我们要找的中位数的坐标,如果是偶数个数,则就是求他们和的一半,如果是奇数个那就是同一个坐标,即left=right
        int left=(n+m+1)/2;
        int right=(n+m+2)/2;
        return (getK(nums1,0,n-1,nums2,0,m-1,left)+getK(nums1,0,n-1,nums2,0,m-1,right))*0.5;
    }
    public int getK(int[] nums1,int start1,int end1,int[] nums2,int start2,int end2,int k){
        // 求的是数组没被排除所剩的长度
        int len1=end1-start1+1;
        int len2=end2-start2+1;
        // 为了好计算,我们默认第一个数组比第二个数组短,如果第一个数组剩余部分比第二个数组剩余部分长了,我们就先交换一下两个数组的位置
        if(len1>len2){
            return getK(nums2,start2,end2,nums1,start1,end1,k);
        }
        // 如果较短数组中已经没有了元素,则直接第二个数组中直接按顺序获取第k个数
        if(len1==0){
            return nums2[start2+k-1];
        }
        // 如果k=1,就找出两个数组第一个数中的最小数
        if(k==1){
            return Math.min(nums1[start1],nums2[start2]);
        }
        // 如果以上情况都不是,那我们就要开始考虑截取长度(k/2)和当前数组所剩长度的问题,应该取最小值(如果取大的数组会越界)
        int i=start1+Math.min(len1,k/2)-1;
        int j=start2+Math.min(len2,k/2)-1;
        // 如果nums1[i]>nums2[j],则证明nums1[j]包括它之前的数都比nums1[i]小,肯定不是第k小数,则可以直接排除,则nums2下次就从j+1开始,而因为已经排除了j-start2+1个数了,则下一次递归,只需要找出第k-(j-start2+1)小的数了。nums2[j]>nums1[i]同理
        if(nums1[i]>nums2[j]){
            return getK(nums1,start1,end1,nums2,j+1,end2,k-(j-start2+1));
        }else{
            return getK(nums1,i+1,end1,nums2,start2,end2,k-(i-start1+1));
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值