leetcode–34.在排序数组中查找元素的第一个和最后一个位置
题目描述
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。你的算法时间复杂度必须是 O(log n) 级别。
如果数组中不存在目标值,返回 [-1, -1]。
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/
思路
二分法
1.比较常规的解法是,二分法解决有序数组的搜索问题,完成二分搜索后,对重复元素进行左右搜索:
分以下分三类讨论:
1.没有目标元素;
2.目标元素在第一位;
3.目标元素有两个或以上
但是最后在去掉重复元素时,时间复杂度最坏可能是O(n)
class Solution:
def searchRange(self, nums: List[int], target: int) -> List[int]:
l , r = 0, len(nums) - 1
res = -1
while l <= r:
mid = l+ ((r-l) >> 1)
# print(nums[l:r+1],mid)
if nums[mid] == target:
res = mid
break
elif nums[mid] > target:
r = mid - 1
else:
l = mid + 1
# 以下分三类讨论:
if res == -1:#没有目标元素
return [-1,-1]
if res == 0:#目标元素在第一位
e =