34.在排序数组中查找元素的第一个和最后一个位置.go

/*
 * @lc app=leetcode.cn id=34 lang=golang
 *
 * [34] 在排序数组中查找元素的第一个和最后一个位置
 *
 * https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/description/
 *
 * algorithms
 * Medium (36.44%)
 * Likes:    130
 * Dislikes: 0
 * Total Accepted:    18.3K
 * Total Submissions: 50.3K
 * Testcase Example:  '[5,7,7,8,8,10]\n8'
 *
 * 给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
 *
 * 你的算法时间复杂度必须是 O(log n) 级别。
 *
 * 如果数组中不存在目标值，返回 [-1, -1]。
 *
 * 示例 1:
 *
 * 输入: nums = [5,7,7,8,8,10], target = 8
 * 输出: [3,4]
 *
 * 示例 2:
 *
 * 输入: nums = [5,7,7,8,8,10], target = 6
 * 输出: [-1,-1]
 *
 */
func searchRange(nums []int, target int) []int {
	n := len(nums)
	if n <= 0 {
		return []int{-1, -1}
	}
	lastIndex := binarySearch(nums, target)
	if lastIndex == -1 || nums[lastIndex] != target {
		return []int{-1, -1}
	}
	firstIndex := binarySearch(nums, target-1)
	return []int{firstIndex + 1, lastIndex}

}

func binarySearch(nums []int, target int) int {
	lo, hi := 0, len(nums)
	for lo < hi {
		mi := (lo + hi) >> 1
		if nums[mi] <= target {
			lo = mi + 1
		} else {
			hi = mi
		}
	}
	return lo - 1
}