81. 搜索旋转排序数组 II
原始题目链接:https://leetcode-cn.com/problems/search-in-rotated-sorted-array-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
解题思路:
和33题不一样的地方是数组中有相同的元素,这样存在nums[left]==nums[mid]==nums[right]的可能,所以遇到这种的情况的时候,解决的办法是收缩区间,即left向右边移动同时right向左边移动,相当于去重。剩下的和33题是一样的,每次计算完mid总有一半即一侧是有序的,有序的一侧就使用二分查找,无序的再一分为二,分完后会至少保证得到一半是有序的,判断有序的部分使用二分查找,这样持续循环。
代码实现:
class Solution:
def search(self, nums: List[int], target: int) -> bool:
if not nums:
return False
n = len(nums)
l, r = 0, n - 1
while l <= r:
mid = l + (r - l) // 2
if nums[mid] == target:
return True
if nums[l] == nums[mid] == nums[r]:
l += 1
r -= 1
elif nums[l] <= nums[mid]:
if nums[l] <= target < nums[mid]:
r = mid - 1
else:
l = mid + 1
else:
if nums[mid] < target <= nums[r]:
l = mid + 1
else:
r = mid - 1
return False
参考文献:
https://leetcode-cn.com/problems/search-in-rotated-sorted-array-ii/solution/acm-xuan-shou-tu-jie-leetcode-sou-suo-xu-wkah/