LeetCode第十五天——搜索旋转排序数组

题目

假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如，数组 [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 nums==[]:
            return -1
        if len(nums)==1:
            return 0 if target in nums else -1
        if len(nums) in [1,2,3]:
            if target in nums:
                return nums.index(target)
            else:
                return -1
        if nums[0]<nums[len(nums)-1]:#如果有旋转
            return self.search_target(nums,0,len(nums)-1,target)
        else:#如果没旋转
            point=self.search_point(nums,0,len(nums)-1) ##至少三个点
            if target>=nums[0]:
                return self.search_target(nums,0,point-1,target)
            else:
                return self.search_target(nums,point,len(nums)-1,target)
        
        
    def search_point(self,nums,l,r):
        if l==r-1:
            return r if nums[l]>nums[r] else l
        if nums[(l+r)//2-1]>nums[(l+r)//2+1]>nums[(l+r)//2]:
            return (l+r)//2
        if nums[(l+r)//2]>nums[l]:
            return self.search_point(nums,(l+r)//2,r)
        if nums[(l+r)//2]<nums[l]:
            return self.search_point(nums,l,(l+r)//2)

        
    def search_target(self,nums,l,r,target):
        if target<nums[l] or target>nums[r]:
            return -1
        if l==r:
            return -1 if nums[l]!=target else l
        if nums[(l+r)//2]==target:
            return (l+r)//2
        elif nums[(l+r)//2]>target:
            return self.search_target(nums,l,(l+r)//2-1,target)
        else:
            return self.search_target(nums,(l+r)//2+1,r,target)