题目:https://leetcode-cn.com/problems/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 可能包含重复元素。
这会影响到程序的时间复杂度吗?会有怎样的影响,为什么?
解题思路:
二分查找,因数列进行了一次旋转,所以在二分点mid = start + (end - start) // 2的左右两边一定有一边是有序的,只需比较target与有序边的边界值判断是否在该区间。
此题需要注意的是,序列存在相同值,所以会存在[1,1,3,1]这种情况,第一次mid指向了下标1,由于nums[start] == nums[mid] == nums[end],无法判断两边哪边有序,此时需要舍弃相同的值,即当start位与mid位相同时,start++,end位与mid位相同时,end--。
代码:
class Solution:
def search(self, nums: List[int], target: int) -> bool:
start = 0
end = len(nums) - 1
while(start <= end):
mid = start + (end - start) // 2
if(nums[mid] == target):
return True
if(nums[mid] < nums[end]):
if(nums[mid] <= target <= nums[end]):
start = mid + 1
else:
end = mid - 1
elif(nums[start] < nums[mid]):
if(nums[start] <= target <= nums[mid]):
end = mid - 1
else:
start = mid + 1
elif(nums[start] == nums[mid]):
start = start + 1
elif(nums[end] == nums[mid]):
end = end - 1
return False
测试用例:
if __name__=="__main__":
nums = [2,5,6,0,0,1,2]
target = 0
res = Solution().search(nums, target)
print(res)