【二分之美】

当看到有序数组,时间复杂度限定为logN时,便想到二分法。

一、leetcode4寻找两个有序数组的中位数

在这里插入图片描述
题目里要求了时间复杂度为logN,如果不做要求可以使用双指针分别指向两个数组的头部,一直移动到length之和的一半。
我们不妨用另一种思路,题目是求中位数,其实就是求第 k 小数的一种特殊情况,而求第 k 小数有一种算法。
假设我们要找第 7 小的数字。
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
7-7/2=4,4-4/2=2,2-2/2=1,------>7=4+2+1
即k-k/2一直递归下去,直到k=1时比较指针指向元素的大小。但是要处理一下极端情况:当一个数组的指针加k/2大于数组长度是取数组长度。

 int i = Math.min(firstCurr+k/2-1,nums1.length-1);
 int j = Math.min(secondCurr+k/2-1,nums2.length-1);

并且当一个数组的指针已经指向数组结尾时,返回另一数组的第k个元素:

        if(firstCurr==nums1.length)
            return nums2[secondCurr+k-1];
        if(secondCurr==nums2.length)
            return nums1[firstCurr+k-1];

整体代码:

/*
要求时间复杂度log,用二分,转换为求两个有序数组第k个小的数
 */
public class code4 {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        if((nums1==null||nums1.length==0)&&(nums2==null||nums2.length==0))
            return 0.0;
        int k1 = (nums1.length+nums2.length+1)/2;
        int k2 = (nums1.length+nums2.length+2)/2;
        //将偶数和奇数的情况合并,如果是奇数,会求两次同样的 k 。
        int res1 = firstK(nums1, nums2, 0,0,k1);
        int res2 = firstK(nums1, nums2, 0,0,k2);
        return (res1+res2)/2.0;
    }

    private int firstK(int[] nums1, int[] nums2, int firstCurr,
                       int secondCurr, int k) {
        if(firstCurr==nums1.length)
            return nums2[secondCurr+k-1];
        if(secondCurr==nums2.length)
            return nums1[firstCurr+k-1];
        if(k==1)
            return Math.min(nums1[firstCurr],nums2[secondCurr]);
        int i = Math.min(firstCurr+k/2-1,nums1.length-1);
        int j = Math.min(secondCurr+k/2-1,nums2.length-1);
        if(nums1[i]>=nums2[j])
            return firstK(nums1, nums2, firstCurr,j+1,k-(j-secondCurr+1));

        else
            return firstK(nums1, nums2, i+1,secondCurr,k-(i-firstCurr+1));

    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值