给定一个按照升序排列的整数数组 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 {
if len(nums) == 0 {
return []int{-1, -1}
}
//找到头
start := findFirst(nums, target)
if start == -1 {
return []int{-1, -1}
}
//找到尾
end := findLast(nums, target)
return []int{start, end}
}
func findFirst(nums []int, target int) int {
start, end := 0, len(nums)-1
for start < end {
mid := (start + end) / 2
if nums[mid] > target {
end = mid - 1
} else if nums[mid] == target {
end = mid
} else {
start = mid + 1
}
}
//此处防止数组第一个数是 target ,所以校验下是否相等
if nums[start] == target {
return start
}
return -1
}
func findLast(nums []int, target int) int {
start, end := 0, len(nums)-1
for start < end {
//此处注意,为了防止 start=mid<end 导致死循环的问题
mid := (start + end + 1) / 2
if nums[mid] > target {
end = mid - 1
} else if nums[mid] == target {
start = mid
} else {
start = mid + 1
}
}
return end
}