天天看點

《leetcode-go》求目标值的區間

給出一個有序數組,請在數組中找出目标值的起始位置和結束位置

你的算法的時間複雜度應該在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}
}