一、题设
已知存在一个按非降序排列的整数数组 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.若nums[mid] == nums[left]:此时mid下标处和第一个数相同,比较不了哪个数组是有序的,从而使left自加跳过这个数判断下个数。例如nums = [2,3,2,2,2],此时应是后半段[mid,right]有序,那么跳到3比较,转3.
2.若nums[mid] > nums[left]:此时前半段[left,mid]有序,再去判断target是否在这个范围,若在则right = mid -1(去前半段找);若不在,则left = mid + 1(去后半段找).
3.若nums[mid] < nums[left]:此时后半段[mid,right]有序,再去判断target是否在这个范围,若在则left = mid +1(去后半段找);若不在,则right = mid - 1(去前半段找).
这题同Topic33搜索旋转排序数组.
三、代码实现
class Solution(object):
def search(self, nums, target):
left , right = 0 ,len(nums)-1
while left <= right:
mid = (left + right) // 2
if nums[mid] == target:
return True
if nums[left] == nums[mid]:#判断不了前后哪个有序
left += 1
continue
elif nums[mid] > nums[left]:# 前半截有序
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
四、效率总结