leetcode81.搜索旋转排序数组 II

81. 搜索旋转排序数组 II

中等（二分查找）

已知存在一个按非降序排列的整数数组 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,4,4,5,6,6,7] 在下标 5 处经旋转后可能变为 [4,5,6,6,7,0,1,2,4,4] 。

给你 旋转后 的数组 nums 和一个整数 target ，请你编写一个函数来判断给定的目标值是否存在于数组中。如果 nums 中存在这个目标值 target ，则返回 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

代码：二分查找

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
class Solution:
    def search(self, nums: List[int], target: int) -> bool:
        left, right = 0, len(nums) - 1
        while left <= right:
            mid = (left + right) // 2
            if nums[mid] == target:
                return True
            # 当无法判断哪部分有序时，缩小两端
            if nums[left] == nums[mid] == nums[right]:
                left += 1
                right -= 1
            elif nums[left] <= nums[mid]:  # 左半部分有序
                if nums[left] <= target < nums[mid]:
                    right = mid - 1
                else:
                    left = mid + 1
            else:  # 右半部分有序
                if nums[mid] < target <= nums[right]:
                    left = mid + 1
                else:
                    right = mid - 1
        return False

思路：

由于数组是旋转过的非降序的数组，所以在使用二分查找目标值的过程中可能会出现三种情况：第一种：此时中间值与左右两端值都一样，无法判断左右两部分是否有序，那就扔掉左右两个端点，重新二分查找。第二种：左半部分有序，我们可以判断目标值是否在左半部分，如果在，右端点移到中间位置的左边；如果不在，左端点移到中间位置的右边。第三种：右半部分有序，操作与第二种类似。