Python描述 LeetCode 33. 搜索旋转排序数组

Python描述 LeetCode 33. 搜索旋转排序数组

  大家好，我是亓官劼（qí guān jié ），在【亓官劼】公众号、、GitHub、B站等平台分享一些技术博文，主要包括前端开发、python后端开发、小程序开发、数据结构与算法、docker、Linux常用运维、NLP等相关技术博文，时光荏苒，未来可期，加油~

  如果喜欢博主的文章可以关注博主的个人公众号【亓官劼】(qí guān jié)，里面的文章更全更新更快。如果有需要找博主的话可以在公众号后台留言，我会尽快回复消息.

本文原创为【亓官劼】（qí guān jié ），请大家支持原创，部分平台一直在恶意盗取博主的文章！！！ 全部文章请关注微信公众号【亓官劼】。

题目

整数数组 ​​nums​​ 按升序排列，数组中的值 互不相同 。

在传递给函数之前，​​nums​​​ 在预先未知的某个下标 ​​k​​​（​​0 <= k < nums.length​​）上进行了 旋转，使数组变为 ​​[nums[k], nums[k+1], ..., nums[n-1], nums[0], nums[1], ..., nums[k-1]]​​（下标 从 0 开始 计数）。例如， ​​[0,1,2,4,5,6,7]​​​ 在下标 ​​3​​​ 处经旋转后可能变为 ​​[4,5,6,7,0,1,2]​​ 。

给你 旋转后 的数组 ​​nums​​​ 和一个整数 ​​target​​​ ，如果 ​​nums​​​ 中存在这个目标值 ​​target​​​ ，则返回它的下标，否则返回 ​​-1​​ 。

示例 1：

输入：nums = [4,5,6,7,0,1,2], target = 0
输出：4

示例 2：

输入：nums = [4,5,6,7,0,1,2], target = 3
输出：-1

示例 3：

输入：nums = [1], target = 0
输出：-1

提示：

• ​​1 <= nums.length <= 5000​​
• ​​-10^4 <= nums[i] <= 10^4​​
• ​​nums​​ 中的每个值都独一无二
• 题目数据保证​​nums​​ 在预先未知的某个下标上进行了旋转
• ​​-10^4 <= target <= 10^4​​

解题思路

先找到第一个逆序的点，即旋转的点。将序列分为两段，分别二分即可。

Python描述

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        idx, n = 0, len(nums)
        # 找第一个逆序的位置
        while idx < n-1:
            if nums[idx] > nums[idx+1]:
                break
            idx += 1
        
        # 在两段升序序列中二分查找target
        l, r = 0, idx
        while l <= r:
            mid = l + r>> 1
            if nums[mid] == target:
                return mid
            elif nums[mid] < target:
                l = mid + 1
            else:
                r = mid - 1

        l, r = idx+1, n-1
        while l <= r:
            mid = l + r>> 1
            if nums[mid] == target:
                return mid
            elif nums[mid] < target:
                l = mid + 1
            else:
                r = mid - 1

        return -1