题目
题目详情
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
你的算法时间复杂度必须是 O(log n) 级别。
如果数组中不存在目标值,返回 [-1, -1]。
示例
输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]
输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]
解题思路
思路
- 这道题很简单,只需要把数组中等于目标值的数的位置依次存到新数组中
- 那么步骤就是先遍历,再判断,最后存到新数组中即可
- 输出的数据为新数组的第一个元素和最后一个元素(如果新数组为空,那么返回[-1,-1]即可)
代码
class Solution:
def searchRange(self,nums,target):
list1 = []
for i in range(len(nums)): #遍历
if nums[i] == target: #是否等于目标值
list1.append(i) #把位置存到新数组
if list1: #判断目标值是否存在数组中
return [list1[0],list1[-1]] #返回新数组的第一个元素和最后一个元素
else:return [-1,-1]
运行结果
最佳方案
我认为我这种思想比较简单,但是需要额外的空间。这位大神的用时36ms,可以参考一下,学习另外一种方法,这种方法是
class Solution:
def binaryFirstSearch(self, nums, target):
if len(nums) == 0:
return -1
start = 0
end = len(nums) - 1
while start <= end:
mid = (start + end)//2
if nums[mid] > target:
end = mid - 1
elif nums[mid] == target:
end = mid - 1
else :
start = mid + 1
if start < len(nums) and nums[start] == target:
return start
return -1
def binaryLastSearch(self,nums,target):
if len(nums) == 0:
return -1
start = 0
end = len(nums) - 1
while start <= end:
mid = (start + end)//2
if nums[mid] > target:
end = mid - 1
elif nums[mid] == target:
start = mid + 1
else :
start = mid + 1
if end >= 0 and nums[end] == target:
return end
return -1
def searchRange(self, nums, target):
first = self.binaryFirstSearch(nums,target)
end = self.binaryLastSearch(nums,target)
return [first,end]