整数数组 nums 按升序排列,数组中的值 互不相同 。
在传递给函数之前,nums 在预先未知的某个下标 k(0 <= k < nums.length)上进行了 旋转,使数组变为 [nums[k], nums[k+1], ..., nums[n-1], nums[0], nums[1], ..., nums[k-1]](下标 从 0 开始 计数)。例如, [0,1,2,4,5,6,7] 在下标 3 处经旋转后可能变为 [4,5,6,7,0,1,2] 。
给你 旋转后 的数组 nums 和一个整数 target ,如果 nums 中存在这个目标值 target ,则返回它的下标,否则返回 -1 。
你必须设计一个时间复杂度为 O(log n) 的算法解决此问题
分析
- 用到二分查找,另外需要注意的是用mid再次明确所在的区间,然后再用常用的left和right的移动
class Solution:
def search(self, nums: List[int], target: int) -> int:
left, right = 0, len(nums) - 1
while left <= right:
mid = left + (right - left)//2
if nums[mid] == target:
return mid
# 上面是return, 所以下面可以直接用if
# 外界条件有点多,可以减少,将条件放到里面
# 用mid和left的元素判断即可
if nums[left] <= nums[mid]:
# 这个区间是最明确的,升区间
if nums[left] <= target and target < nums[mid]:
right = mid - 1
else:
left = mid + 1
else:
# 这个区间也是最明确的,升区间
if target <= nums[right] and target > nums[mid]:
left = mid + 1
else:
right = mid - 1
return -1