給出一個有序數組,請在數組中找出目标值的起始位置和結束位置
你的算法的時間複雜度應該在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}
}