题目描述

题目链接

给你一个按照非递减顺序排列的整数数组 nums，和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target，返回 [-1, -1]。

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

示例 1：

输入： nums = [5,7,7,8,8,10], target = 8
输出： [3,4]

示例 2：

输入： nums = [5,7,7,8,8,10], target = 6
输出： [-1,-1]

示例 3：

输入： nums = [], target = 0
输出： [-1,-1]

提示：

• 0 <= nums.length <= 105
• -109 <= nums[i] <= 109
• nums 是一个非递减数组
• -109 <= target <= 109

思路

二分查找过程中插入搜索目标值起始位置和终止位置的逻辑即可。

代码

func searchRange(nums []int, target int) []int {
    // 变形二分查找
    var start, end int
    var l, r, mid int
    l, r = 0, len(nums)-1
    for l <= r {
        mid = (r-l)/2 + l
        if nums[mid] == target {
            start, end = mid, mid
            for start >= 0 && nums[start] == target {
                start--
            }
            for end < len(nums) && nums[end] == target {
                end++
            }
            return []int{start+1, end-1}
        }
        if nums[mid] > target {
            r = mid - 1
        } else {
            l = mid + 1
        }
    }
    return []int{-1, -1}
}

图示

略