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

题目链接: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在其中都是可见的

递推阶段的结束情况:

  1. target不在子数组范围内(target小于子数组最小值或大于子数组最大值)
  2. 子数组长度为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
}

你可能感兴趣的:(刷题日常)