题目在这:https://leetcode-cn.com/problems/search-in-rotated-sorted-array/
思路分析:
这道题说了一堆题目描述,实际上就是给了一个数组nums,给了一个数target。从数组nums中找到target这个数就行了。
用if ....in .....
.就能直接做出来这道题,而且还不超时还能通过。
class Solution:
def search(self, nums: List[int], target: int) -> int:
return nums.index(target) if target in nums else -1
既然能通过题目干嘛唠叨这么多东西,又旋转又什么的。
往后看,有个进阶。要求时间复杂度在 logn之内。
搜索数字的话,这个时间复杂度,二分吧。
不过这道题需要注意,题目中唠叨这么多就是想告诉你,给的这个数组是经过旋转的,即 分两半,两半分别有序。
这时候在使用二分法的时候,我们需要对比中间值mid在那一边是有序的,然后再判断目标值再那一边,如果有无序的那一边则无限分割无序的那一边分割成有序的。
由于题目说数字无重复,举个例子:
1 2 3 4 5 6 7 可以大致分为两类,
-
第一类 2 3 4 5 6 7 1 这种,也就是 nums[start] <= nums[mid]。此例子中就是 2 <= 5。
这种情况下,前半部分有序。因此如果 nums[start] <=target<nums[mid],则在前半部分找,否则去后半部分找
。 -
第二类 6 7 1 2 3 4 5 这种,也就是 nums[start] > nums[mid]。此例子中就是 6 > 2。
这种情况下,后半部分有序。因此如果 nums[mid] <target<=nums[end],则在后半部分找,否则去前半部分找。
完整代码
class Solution:
def search(self, nums: List[int], target: int) -> int:
left = 0
right = len(nums) - 1
while left <= right:
mid = (left + right) // 2
if nums[mid] == target: # 找到目标值
return mid
if nums[left] <= nums[mid]: # 左半部分有序
if nums[left] <= target < nums[mid]: # 去左边找
right = mid - 1
else: # 去右边找
left = mid + 1
# nums = [7, 6, 5, 4, 0, 1, 2]
else: # 右半部分有序
if nums[mid] < target <= nums[right]: # 找右
left = mid +1
else: # 找左
right = mid -1
return -1
题目中的等号我也纠结了一阵,调试几次就能知道怎么加了,没必要纠结。