【35】搜索插入位置

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_36172443/article/details/82146134

给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。

你可以假设数组中无重复元素。

示例 1:

输入: [1,3,5,6], 5
输出: 2
示例 2:

输入: [1,3,5,6], 2
输出: 1
示例 3:

输入: [1,3,5,6], 7
输出: 4
示例 4:

输入: [1,3,5,6], 0
输出: 0

思路: 有序=》二分查找

func searchInsert(nums []int, target int) int {
    l,r:=0,len(nums)-1
    var mid int
    for l<=r{
        mid =(l+r)/2
        if(nums[mid]==target){
            return mid
        }
        if(nums[mid]<target){
            l=mid+1
        }else{
            r=mid-1
        }

    }
    if(nums[mid]>target){
        return mid
    }else{
        return mid+1
    }
}
优化版本

func searchInsert(nums []int, target int) int {
    l,r:=0,len(nums)-1
    var mid int
    for l<=r{
        mid =l+(r-l)>>1
        switch{
            case nums[mid]<target:
            l=mid+1
            case nums[mid]>target:
            r=mid-1
            default :
            return mid

        }

    } 
    return l

}

猜你喜欢

转载自blog.csdn.net/qq_36172443/article/details/82146134