题目
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [0,0,1,2,2,5,6] 可能变为 [2,5,6,0,0,1,2] )。
编写一个函数来判断给定的目标值是否存在于数组中。若存在返回 true,否则返回 false。
链接:https://leetcode.com/problems/search-in-rotated-sorted-array-ii/
Suppose an array sorted in ascending order is rotated at some pivot unknown to you beforehand.
(i.e., [0,0,1,2,2,5,6] might become [2,5,6,0,0,1,2]).
You are given a target value to search. If found in the array return true, otherwise return false.
Example:
Input: nums = [2,5,6,0,0,1,2], target = 0
Output: true
思路及代码
二分查找
- 其他思路和Leetcode-33. 搜索旋转排序数组一样,唯一的区别是重复值
- 可能遇到的情况为[1,1,1,1,3,1]和[1,3,1,1,1,1]这样就不知道目标在哪边,所以遇到时将high向左移动一位,然后继续寻找就可以了
class Solution:
def search(self, nums: List[int], target: int) -> bool:
low = 0
high = len(nums) - 1
while low <= high:
mid = int((low+high) / 2)
if nums[mid] == target:
return True
# 如果遇到相同值,则high向左移一位
if nums[mid] == nums[high]:
high = high - 1
elif nums[mid] < nums[high]:
if nums[mid] < target and nums[high] >= target:
low = mid + 1
else:
high = mid - 1
elif nums[mid] > target and nums[low] <= target:
high = mid - 1
else:
low = mid + 1
return False
复杂度
T = O ( l o g n ) O(logn) O(logn)