假设按照升序排序的数组在预先未知的某个点上进行了旋转。( 例如,数组 [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
解题思路:
O(logN)一般都是想到二分,此题目中有旋转点关键在于借助有序的区间进行二分,还要明白一个如果target不在当前的有序区间内怎么办,其实不用特殊处理,因为这种情况与原题目中的情况是一样的,理解为一个子问题就好。也就是说,每次mid划分后,不管是左边还是右边,必然会存在一个有序区间。注意各个区间的开闭等号的写法,不能忽略细节。
class Solution:
def search(self, nums: List[int], target: int) -> int:
'''
O(logN)一般都是想到二分,此题目中有旋转点
关键在于借助有序的区间进行二分,还要明白一个
如果target不在当前的有序区间内怎么办,其实不用特殊处理,
因为这种情况与原题目中的情况是一样的,理解为一个子问题就好
'''
if not nums:
return -1
left,right = 0,len(nums)-1
while left < right:
mid = left + (right - left)//2
if nums[mid] == target:
return mid
if nums[mid] >= nums[left]: # 左侧left->mid区间有序
if nums[left] <= target < nums[mid]:
right = mid - 1
else:
left = mid + 1
else: # 很关键,这里的意思其实是右侧有序,mid->right区间
if nums[mid] < target <= nums[right]:
left = mid + 1
else:
right = mid - 1
if nums[left] == target:
return left
else:
return -1
网友评论