【代码随想录】35. 搜索插入位置

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

请必须使用时间复杂度为 O(log n) 的算法。

示例 1:

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

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

输入: nums = [1,3,5,6], target = 7
输出: 4

使用二分法就可以很快解决了,需要判断的是究竟return哪个?

 if target == nums[mid]

这种情况就可以直接return mid了

另外的条件判断target > nums[mid]  和   target < nums[mid]的情况

1.当target在左区间,我们需要right-1,举个例子1,3,5,7,9 mid=2 nums[2]=5,我们需要插入target=4

right=mid-1 所以right=1,那么这种情况我们需要最后return right+1 就是target需要插入的位置.

2.当target在右区间,我们需要插入target=6 nums[2]=5, target在左区间, left=mid+1=3,right=4

mid=(3+4)/2=3, nums[3]=7>target=6, target又回到了左区间,right=mid-1=2 right<left,结束while。

这个时候就知道6放在5后面, return right+1

3.可以跑一次放在第一位和最后一位也是right+1

4.所以最后return right+1即可

func searchInsert(nums []int, target int) int {
    left,right:=0,len(nums)-1
    for left<=right {
        mid:=left+(right-left)>>1
        if target == nums[mid]{
            return mid
        } else if target > nums[mid] {
            left=mid+1
        } else {
            right=mid-1
        }
    }

      // 分别处理如下四种情况
      // 目标值在数组所有元素之前  [0, -1]
      // 目标值等于数组中某一个元素  return middle;
      // 目标值插入数组中的位置 [left, right],return  right + 1
     // 目标值在数组所有元素之后的情况 [left, right], 因为是右闭区间,所以 return right + 1
    return right+1
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值