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)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章