已知存在一个按非降序排列的整数数组 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 。
来源:力扣(LeetCode)
链接: https://leetcode-cn.com/problems/search-in-rotated-sorted-array-ii
这是LeetCode.33搜索旋转排序数组的升级版,数据变成可能有重复的。
LeetCode.33 搜索旋转排序数组题解_Hello World71的博客-CSDN博客(这是之前题目的链接)。
我们现在要思考,变化在哪里?可不可以直接用之前的模板?
大体上都是正确的,只有一个细节要注意:
else if(nums[mid] >= nums[l])
{
if(target < nums[mid] && target >= nums[l])
r = mid -1;
else
l = mid + 1;
}
这是我们之前的代码,num[mid] >= num[l] ,就可以保证 mid 与 l 在同一侧,但是现在不行了,比如下面这种情况:
目标被跳过了,故我们要单独讨论这种情况,
if(nums[mid] == nums[l])
l++;
continue;
只需要单独判断一下,然后将最左边的元素加1,再循环判断。
class Solution {
public:
bool search(vector<int>& nums, int target) {
int l = 0,r = nums.size()-1;
while(l <= r)
{
int mid = l + ((r-l)>>1);
if(nums[mid] == target)
return true;
if(nums[mid] == nums[l])
{ l++;continue;
}
if(nums[mid] > nums[l])
{
if(nums[mid] > target && target >= nums[l])
r = mid - 1;
else
l = mid + 1;
}
else
{
if(target < nums[l] && target > nums[mid])
l = mid + 1;
else
r = mid - 1;
}
}
return false;
}
};