假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。
搜索一个给定的目标值,如果数组中存在这个目标值,则返回它的索引,否则返回 -1 。
你可以假设数组中不存在重复的元素。
你的算法时间复杂度必须是 O(log n) 级别。
示例 1:
输入: nums = [4,5,6,7,0,1,2], target = 0
输出: 4
示例 2:
输入: nums = [4,5,6,7,0,1,2], target = 3
输出: -1
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-in-rotated-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
使用二分法达到要求的时间复杂度
1.先使用二分法求最大值所在,并标记
2.考虑特殊情况。长度为0/1的情况 / middle在首位的情况、
3.一般情况。最大值所在位置与target比较,确定二分法寻找的范围
class Solution:
def search(self, nums: List[int], target: int) -> int:
#求分界线
def find_middle(left,right):
if nums[left]<nums[right]:
return 0
while left<=right:
middle = (left+right)//2
if nums[middle]>nums[middle+1]:
return middle+1
else:
if nums[middle] < nums[left]:
right = middle-1
else:
left = middle+1
#在一定范围内使用二分,达到要求时间复杂度
def search_tar(left, right):
while left <= right:
pivot = (left + right) // 2
if nums[pivot] == target:
return pivot
else:
if target < nums[pivot]:
right = pivot - 1
else:
left = pivot + 1
return -1
l = len(nums)
left = 0
right = l-1
#长度特殊情况
if l==0:
return -1
elif l==1:
return 0 if nums[0]==target else -1
middle=find_middle(0,l-1)
#一般情况,使用middle位置与target的比较
if target==nums[middle]:
return middle
elif middle==0:
return search_tar(0,l-1)
elif target<nums[0]:
return search_tar(middle,l-1)
return search_tar(0,middle)