原题:
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [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
首先,我们采用直接的查找方法:
def search(self, nums: list, target: int) -> int:
i = len(nums)
for j in range(0,i):
if nums[j]==target:
return j
return -1
self = []
nums=[4,5,6,7,0,1,2]
target = 3
print(search(self,nums,target))
效果:
接下来,我们采用二分方法:开始时定义left、right指针,将left指向开头,right指向末尾;与target比较;如果nums[left]<target,就将left指针加1;如果nums[right]<target,就将right指针减1;如果不满足以上两种情况,就返回-1;另外,判断nums[left]是否小于nums[left-1],如果是,就调用二分法查找left到right的范围;如果nums[right]>nums[right-1],调用二分法查找left到right的范围
代码:
二分法:
def half(nums:list,target:int)->int:
i = len(nums)
left = 0
right = i-1
while left<=right:
s = int((left+right)/2)
mid = nums[s]
if mid ==target:
return s
elif mid <target:
left = s+1
else:
right=s-1
return -1
函数:
def search(self, nums: list, target: int) -> int:
i = len(nums)
left = 0
right = i-1
while left<right:
if target == nums[left]:
return left
if target==nums[right]:
return right
if target>nums[left]:
left = left+1
if nums[left]<nums[left-1]:
half(nums[left:right],target)
break
elif target<nums[right]:
right = right-1
if nums[right]>nums[right+1]:
half(nums[left:right],target)
break
else:
return -1
return -1
self = []
nums=[4,5,6,7,0,1,2]
#nums = [0,1,2,3,4,5]
target = 5
print(search(self,nums,target))
效果: