- 题目描述
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [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
- 思路
看了评论和题解才明白到底是什么意思,给出的这个数组可能是旋转过的,所以在进行查找之前,如果能把这个数组分成两个升序数组分别来找,肯定是比较好的解决办法。这两部分都可以用二分法来处理。
将原数组分成两个升序数组的操作也可以通过二分查找来实现,通过比较中间值与最大值/最小值的大小,移动比较范围的最小、最大索引,最终确定旋转的中心点,也就是整个数组中的最小值,同时也是两部分的分界点。
- 代码
class Solution:
def search(self, nums: List[int], target: int) -> int:
if not nums: return -1
i,j = 0,len(nums)-1
#找到最小值
while i < j:
m = (i+j)//2
if nums[m] < nums[j]:
j = m
else:
i = m +1
#以i为界限分开nums数组为两个升序数组,看target是否在数组中
if nums[i] <= target <= nums[-1]:
i,j = i,len(nums)-1
elif nums[0] <= target <= nums[i-1]:
i,j = 0,i-1
else:
return -1
#二分查找
while i <= j:
m = (i+j)//2;
if nums[m] > target:
j = m-1
elif nums[m] < target:
i = m + 1
else:
return m
return -1