天天看点

《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}
}