[LeetCode]81. 搜索旋转排序数组 II（Search in Rotated Sorted Array II）

题目描述

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

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

编写一个函数来判断给定的目标值是否存在于数组中。若存在返回 true，否则返回 false。

示例 1:

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

示例 2:

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

进阶:

这是 搜索旋转排序数组 的延伸题目，本题中的 nums 可能包含重复元素。
这会影响到程序的时间复杂度吗？会有怎样的影响，为什么？

解题思路

与[LeetCode]33. 搜索旋转排序数组（Search in Rotated Sorted Array）一样使用二分，如果遇到不能判断左右半边哪边有序时，直接递归化为子问题即可。

代码

Python 3.6

class Solution:
    def search(self, nums: List[int], target: int) -> bool:
        length = len(nums)
        if length == 0:
            return False
        if length == 1:
            return nums[0] == target
        left, right = 0, length - 1
        while left < right:
            mid = (left + right) >> 1
            if nums[left] == nums[mid] == nums[right]:
                # 递归解决左右两半边的子问题
                return self.search(nums[:mid], target) or self.search(nums[mid + 1:], target)
            if nums[left] <= nums[mid]:
                # 左半边单调不减
                if nums[left] <= target <= nums[mid]:
                    right = mid
                else:
                    left = mid + 1
            else:
                # 右半边单调不减
                if nums[mid + 1] <= target <= nums[right]:
                    left = mid + 1
                else:
                    right = mid
        return nums[left] == target

执行用时 : 96 ms
内存消耗 : 14.1 MB