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

33. 搜索旋转排序数组


题目来源:https://leetcode-cn.com/problems/search-in-rotated-sorted-array/python

题目


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

( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。数组

搜索一个给定的目标值,若是数组中存在这个目标值,则返回它的索引,不然返回 -1 。bash

你能够假设数组中不存在重复的元素。微信

你的算法时间复杂度必须是 O(log n) 级别。code

示例 1:blog

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

示例 2:排序

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

解题思路


思路:二分查找索引

题目中,要求算法时间复杂度必须是 O(log n) 级别,在这里,其实提示能够使用二分查找的方法。leetcode

有个地方须要注意的是,二分查找,要求的是列表的元素必须是有序的。在这里,虽然原始数组是按照升序排序的,但数组在某个未知的点进行旋转以后,列表只有局部才是有序的。

但其实这里并不会冲突,由于将通过旋转的列表分割成两部分时,必定有一个部分是有序的。

直接以上面的示例 1 为例,[4, 5, 6, 7, 0, 1, 2],若是将数组分割成 [4, 5, 6][7, 0, 1, 2],那么前面 [4, 5, 6] 这部分是有序的。这在其余符合这些条件的数组同样是成立的。

上面提到数组分为两部分,那么能够考虑从在中点的位置进行划分,将原数组划分为 [left, mid][mid + 1, right] 两部分。

当将旋转后的数组分红上述两部分时,如今要考虑的就是 target 值,会落在哪一个区间?当 target 值落在有序那部分的区间中,这个比较好理解,这个时候上下边界是肯定的,题目中也说明【能够假设数组中不存在重复的元素。】,那么有且只有一个元素与之匹配。

如今具体分析判断 target 会落在哪一个区间:

  • 若是 [left, mid - 1] 这部分是有序的,且 target 落在 [nums[left], nums[mid]) 这个区间,那么就能够将搜索的范围缩小到 [left, mid - 1],不然的话,将在 [mid + 1, right] 这个区间中进行搜索;
  • 若是 [mid, right] 这部分是有序的,且 target 落在 (nums[mid+1], nums[right]],那么就在 [mid + 1, right] 这个区间进行搜索,不然在 [left, mid - 1] 这个区间进行搜索。

下面是具体的代码实现。

代码实现


class Solution:
    def search(self, nums: List[int], target: int) -> int:
        # nums 为空时,直接返回 -1
        if not nums:
            return -1

        # 初始左右边界
        left, right = 0, len(nums) - 1
        
        # 判断哪部分区间是有序的
        # 判断 target 会落在哪一个区间
        while left <= right:
            # 数组划分为两部分
            mid = (left + right) // 2
            # 若是目标值与划分区间的元素相等时,直接返回
            if nums[mid] == target:
                return mid

            # 判断区间是否有序
            # 这里表示 [left, mid - 1] 是有序的
            if nums[0] <= nums[mid]:
                # target 落在 [left, mid - 1] 这个区间
                if nums[0] <= target < nums[mid]:
                    # 从新划分右边界
                    right = mid - 1
                # target 落在 [mid + 1, right] 这个区间
                else:
                    # 从新划分左边界
                    left = mid + 1
            
            # nums[0] > nums[mid] 时,表示 [mid + 1, right] 有序
            else:
                # target 落在 [mid + 1, right] 时
                if nums[mid] < target <= nums[right]:
                    # 从新划分左边界
                    left = mid + 1
                else:
                    # target 落在 [left, mid - 1] 这个区间
                    # 从新划分左边界
                    right = mid - 1

        return -1

实现结果


实现结果


以上就是使用二分查找的思想,解决《33. 搜索旋转排序数组》问题的主要内容。


欢迎关注微信公众号《书所集录》

相关文章
相关标签/搜索