假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [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]的7, 特点是这一位比后一位大。
找到之后数组就分成了两段单调递增的区间,将target跟nums[0]比较之后可以判断出target落在哪段区间上,
然后就是普通的二分查找。
class Solution(object): def search(self, nums, target): """ :type nums: List[int] :type target: int :rtype: int """ if not nums: return -1 if len(nums) == 1: return 0 if nums[0] == target else -1 lo, hi = 0, len(nums) - 1 while(lo <= hi): mid = (lo + hi) // 2 # print mid, nums[mid] if mid + 1 < len(nums) and nums[mid] > nums[mid +1]: break if nums[mid] < nums[-1]: hi = mid - 1 elif nums[mid] >= nums[0]: lo = mid + 1 if lo > hi:#没有旋转 lo, hi = 0, len(nums) - 1 else: if target >= nums[0]: lo, hi = 0, mid else: lo, hi = mid + 1, len(nums) - 1 while(lo <= hi): # print lo, hi mid = (lo + hi) // 2 if nums[mid] == target: return mid elif nums[mid] > target: hi = mid - 1 else: lo = mid + 1 return -1