LintCode 搜索区间

给定一个包含 n 个整数的排序数组,找出给定目标值 target 的起始和结束位置。

如果目标值不在数组中,则返回[-1, -1]

样例
给出[5, 7, 7, 8, 8, 10]和目标值target=8,

返回[3, 4]

挑战
时间复杂度 O(log n)

分别对左右边界进行二分查找。注意特殊情况的处理。直接看代码:

public class Solution {
    /*
     * @param A: an integer sorted array
     * @param target: an integer to be inserted
     * @return: a list of length 2, [index1, index2]
     */
    public int[] searchRange(int[] A, int target) {
        // write your code here
        int[] res={-1,-1};
        if(A.length==0){//数组为空,直接返回没找到
            return res;
        }
        //查找右边界
        int l=0,r=A.length-1;
        if(A[A.length-1]==target){//最后一个数等于target,直接设置
            res[1]=A.length-1;
        }
        else{
            while(l<=r){
                int mid=l+(r-l)/2;
                if(A[mid]==target){
                    if(A[mid+1]==target){
                        l=mid+1;
                    }
                    else{
                        res[1]=mid;
                        break;
                    }
                }
                else if(A[mid]<target){
                    l=mid+1;
                }
                else{
                    r=mid-1;
                }
            }
        }
        if(res[1]==-1){//如果右边界没找到,那说明没有这个数,直接返回没找到
            return res;
        }
        //查找左边界,右端起始位置直接设置为刚刚找到的右边界。
        l=0;r=res[1];
        if(A[0]==target){//第一个数等于target,直接设置
            res[0]=0;
        }
        else{
            while(l<=r){
                int mid=l+(r-l)/2;
                if(A[mid]==target){
                    if(A[mid-1]==target){
                        r=mid-1;
                    }
                    else{
                        res[0]=mid;
                        break;
                    }
                }
                else if(A[mid]<target){
                    l=mid+1;
                }
                else{
                    r=mid-1;
                }
            }
        }
        return res;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值