leetcode--在排序数组中查找元素的第一个和最后一个位置--python

题目

题目详情

给定一个按照升序排列的整数数组 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. 这道题很简单，只需要把数组中等于目标值的数的位置依次存到新数组中
2. 那么步骤就是先遍历，再判断，最后存到新数组中即可
3. 输出的数据为新数组的第一个元素和最后一个元素（如果新数组为空，那么返回[-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]