已知存在一个按非降序排列的整数数组 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.length <= 5000
-10^4 <= nums[i] <= 10^4
- 题目数据保证
nums
在预先未知的某个下标上进行了旋转 -10^4 <= target <= 10^4
进阶:
- 这是 搜索旋转排序数组 的延伸题目,本题中的 nums 可能包含重复元素。
这会影响到程序的时间复杂度吗?会有怎样的影响,为什么?
解答
在排序的数组中查找给定数字,所以能直观地想到使用二分搜索,由于数组经过旋转后,只有部分有序,所以可以分别对左右有序的部分进行二分搜索:
class Solution {
public:
bool search(vector<int>& nums, int target) {
if(nums.empty())
return false;
if(nums.size() < 2)
return nums[0] == target;
int l = 0, r = nums.size() - 1;
while(l <= r){
int mid = (l + r) >> 1;
if(nums[mid] == target)
return true;
// 数组可能有重复
if(nums[mid] == nums[l] && nums[mid] == nums[r]){
l++;
r--;
continue;
}
// 左边有序
if(nums[l] <= nums[mid]){
if(nums[l] <= target && target < nums[mid])
r = mid - 1;
else
l = mid + 1;
}
// 右边有序
else{
if(target > nums[mid] && target <= nums[r])
l = mid + 1;
else
r = mid - 1;
}
}
return false;
}
};