LeetCode 33 搜索旋转排序数组 Python

LeetCode 33 搜索旋转排序数组

题目描述：

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

( 例如，数组 [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

思路如下：

• 设置三个指针分别指向list的头、中、尾
• 考虑特殊情况，即list中只有一个值时的情况
• 头指针小于尾指针时执行循环
• 考虑若头指针紧挨着尾指针时的情况（此时mid==left or mid==right），根据不同情况返回值
• 根据target是否等于头、中、尾三个指针所指向的值。若是，返回结果
• 下面利用二分法缩短搜索范围：首先判断头指针指向数值至中指针指向数值是否构成升序序列。若是，继续判断target是否在这个升序序列中；若否，则说明旋转点位于mid的左侧（此时mid至right构成升序序列），进而判断target是否在mid至right的升序序列中

代码如下：

class Solution(object):
    def search(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        left=0
        right=len(nums)-1
        
        if len(nums)==1 and nums[0]==target:
            return 0
        while left<right:
            mid=(left+right)/2
            if right-left==1:
                if target==nums[left]:
                    return left
                if target==nums[right]:
                    return right
                else:
                    return -1
            if target==nums[left]:
                return left
            if target==nums[right]:
                return right
            if target==nums[mid]:
                return mid
            if nums[mid]>nums[left]:
                if target>nums[left] and target<nums[mid]:
                    right=mid-1
                else:
                    left=mid+1
            else:
                if target>nums[mid] and target<nums[right]:
                    left=mid+1
                else:
                    right=mid-1
        return -1