33. 搜索旋转排序数组 Python

2019-05-18  本文已影响0人  小小尧

思路:首先了解到这道题要求的解法是log(n)级别。log(n)与什么最像?当然是二分的思想!但是很多朋友可能会说,排序数组的二分查找我很熟,但这道题可能经过了旋转啊。没错,这数组经过旋转后,形成了2段递增的序列。我们得先找到2个序列的分界点,将数组分成2部分分别二分查找。那么这道题的难题就成功转移到如何找到那个分界点。可能很多朋友又会说,直接从后或者从前按顺序找,当突然前面的数字比当前大就说明这是断点嘛!但题目要求logn,按顺序找的话是n。


作者:我喝酸奶不舔盖
来源:CSDN
原文:https://blog.csdn.net/weixin_41958153/article/details/80862566
版权声明:本文为博主原创文章,转载请附上博文链接!

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        #先找到两个第二个升序数组的第一项的index
        l = 0
        r = len(nums) -1
        while l < r:
            mid = (l + r)//2
            if nums[mid] > nums[r]:
                l = mid + 1
            else:
                r = mid
        pol = l
        ans = self.binary_search(target, nums[:pol])
        if ans == -1:
            ans = self.binary_search(target, nums[pol:])
            if ans != -1:
                ans += len(nums[:pol])
 
        return ans
    
    def binary_search(self, target, nums):
        index = -1
        l = 0
        r = len(nums) - 1
        while l <= r:
            mid = (l+r)//2
            if nums[mid] < target:
                l = mid + 1
            elif nums[mid] > target:
                r = mid - 1
            else:
                index = mid
                break
        return index
图片.png
上一篇下一篇

猜你喜欢

热点阅读