[python刷题]leetcode33.搜索旋转排序数组Search in Rotated Sorted Array

题目

假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。
搜索一个给定的目标值,如果数组中存在这个目标值,则返回它的索引,否则返回 -1 。
你可以假设数组中不存在重复的元素。
你的算法时间复杂度必须是 O(log n) 级别。

思路

二分查找
设置三个索引left right mid,以mid将list切分成凉快,判断哪部分是有序的,在有序切片中比较首尾元素,可以判断target在mid左边还是右边,然后移动left right相应位置

代码

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        left = 0
        right = len(nums)-1
        while left <= right:
            
            mid = (left + right) // 2
            #print(left, right, mid)
            if target == nums[mid]:
                #print (mid)
                return mid
            if (nums[left] <= nums[mid]):
                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

你可能感兴趣的:(leetcode)