【题目】
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [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
【分析过程】
以题目中给的数组[4,5,6,7,0,1,2]为例,我首先想到的是二分查找,但是之前学习的二分查找的前提是有序数组,所以这道题不能直接套用二分查找的代码,但是可以套用二分查找的思想。
查找目标值target是否在数组[4,5,6,7,0,1,2]中有以下几种情况:
- nums[mid] == target,那么我们返回mid即可
- 如果数组中mid左边为递增数列,也就是从nums[left]到nums[mid]递增,那么又可以分为两种情况:
- 1. target在mums[left]和nums[mid]之间 ——我们就只需要到这个区间中寻找
- 2.target不在之间,也就是在另外一个区间中
- 如果数组中mid右边为递增数列,也就是从nums[mid]到nums[right]递增,那么又可以分为两种情况:
- 1. target在mums[mid]和nums[right]之间
- 2.target不在之间,也就是在另外一个区间中
这个方法很好理解,但是要注意边界条件的判断
【性能分析】
时间复杂度:
空间复杂度:
【python代码】
class Solution:
def search(self, nums: list, target: int) -> int:
if len(nums) == 0:
return -1
if len(nums) == 1:
return 0 if nums[0] == target else -1
left = 0
right = len(nums) - 1
while left < right:
mid = left + (right - left) // 2
if nums[mid] == target:
return mid
elif nums[mid] < nums[right]:
if nums[mid] < target <= nums[right]:
left = mid + 1
else:
right = mid - 1
else:
if nums[left] <= target < nums[mid]:
right = mid - 1
else:
left = mid + 1
return left if nums[left] == target else -1
s = Solution()
nums = [4,5,6,7,0,1,2]
target = 3
print(s.search(nums,target))