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

103 阅读1分钟

题目:
给你一个按照非递减顺序排列的整数数组 nums,和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target,返回 [-1, -1]

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。
算法:
方法一:二分查找
二分查找的关键在于left左侧和right右侧的数符合什么条件

func searchRange(nums []int, target int) []int {
    left := lowBound(nums, target)
    if left == len(nums) || nums[left] != target {
        return []int{-1, -1}
    }
    right := lowBound(nums, target + 1) - 1
    return []int{left, right}

}

// 某个数字第一次出现的位置
func lowBound(nums []int, target int) int {
    // 关键在于left左侧和right右侧的数符合什么条件
    // left左侧小于target, right右侧大于等于target
    left, right := 0, len(nums) - 1
    for left <= right {
        mid := left +  (right - left) / 2
        if nums[mid] < target {
            left = mid + 1
        } else {
            right = mid - 1
        }
    }
    // right + 1
    return left
}