腾讯34-搜索旋转排序数组

腾讯34-搜索旋转排序数组leetcode33

假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如，数组 [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 len(nums)==0:return -1
        else:
            start,end=0,len(nums)-1
            while start<=end:
                if start==end:
                    return start if nums[start]==target else -1
                if start==end-1:
                    if nums[start]==target :return start
                    elif nums[end]==target:return end
                    else: return -1
                else :
                    mid=start+(end-start)//2
                    if nums[mid]==target:return mid
                    elif nums[start]<nums[mid]:#前面有序
                        if nums[start]<=target and target<=nums[mid]:
                            end=mid
                        else :
                            start=mid
                    else:#后面有序
                        if nums[mid]<=target and target<=nums[end]:
                            start=mid
                        else:
                            end=mid