给出一个有序数组,请在数组中找出目标值的起始位置和结束位置
你的算法的时间复杂度应该在O(log n)之内
如果数组中不存在目标,返回[-1, -1].
例如:
给出的数组是[50, 70, 70, 80, 80, 100],目标值是80,
返回[3, 4].
思路:二分查找找到目标值,然后左右遍历,获取开始和结束
package main
/**
*
* @param A int整型一维数组
* @param target int整型
* @return int整型一维数组
*/
func searchRange( A []int , target int ) []int {
// write code here
right := len(A) - 1
left := 0
for left <= right {
mid := (left + right) / 2
if A[mid] == target {
//找前后的位置
resLeft := mid - 1
for resLeft >= 0 && A[resLeft] == A[mid] {
resLeft--
}
resLeft++
resRight := mid + 1
for resRight <= len(A)-1 && A[resRight] == A[mid] {
resRight++
}
resRight--
return []int{resLeft, resRight}
}
if A[mid] > target {
right = mid - 1
} else {
left = mid + 1
}
}
return []int{-1, -1}
}