题目链接:https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/
题解链接:https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/solution/qiao-yong-gohan-shu-zi-mian-liang-er-fen-cha-zhao-/
来源:力扣(LeetCode) 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
你的算法时间复杂度必须是 O(log n) 级别。
如果数组中不存在目标值,返回 [-1, -1]。
————————————————————————————
二分查找简单思路
输入数组长度为0时直接返回{-1,-1}
其余情况进入递归函数处理。
定义res为结果数组,seq代表结果数组的下标,初始化为0。用函数字面量在searchRange中定义递归函数,使seq,res,nums,target在其中都是可见的。
递推阶段的结束情况:
- target不在子数组范围内(target小于子数组最小值或大于子数组最大值)
- 子数组长度为1,且元素等于target。赋值给res[seq]。
子数组长度大于1时,分为两部分进入新的递归函数。
第一次找到目标值时seq=0,即赋值给res[0],然后使seq=1;之后找到的目标值只在seq=1中更新。
(注意第一次找到目标值时使res[0],res[1]赋值为同一值,对应数组内只有一个target的情况)
func searchRange(nums []int, target int) []int {
if len(nums) == 0 {
return []int{-1, -1}
}
res := []int{-1, -1}
seq := 0
var Partition func(lo, hi int)
Partition = func(lo, hi int) {
if nums[hi] < target || nums[lo] > target {
return
}
if hi-lo == 0 && nums[lo] == target {
res[seq] = lo
if seq == 0 {
seq = 1
res[seq] = lo
}
}
if hi-lo > 0 {
Partition(lo, (lo+hi)/2)
Partition((lo+hi)/2+1, hi)
}
}
Partition(0, len(nums)-1)
return res
}