LeetCode33. 搜索旋转排序数组(二分法,python)

1. 题目

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

( 例如,数组 [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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-in-rotated-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 题解

判断旋转点在中点的左边还是右边,实现二分。

class Solution:
    def helper(self,nums, start, end, target):
        if end<start:
            return -1
        mid = (start + end)//2
        if nums[mid] == target:
            return mid

        if nums[mid]>nums[end]:
            if nums[mid]>target and (nums[start]<=target):
                return self.helper(nums,start,mid-1,target)
            else:
                return self.helper(nums,mid+1,end,target)
        else:
            if nums[mid]<target and (nums[end]>=target):
                return self.helper(nums,mid+1,end,target)
            else:
                return self.helper(nums,start,mid-1,target)
            
    def search(self, nums: List[int], target: int) -> int:
        return self.helper(nums,0,len(nums)-1,target)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章