一、运行结果
二、题目
已知存在一个按非降序排列的整数数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-in-rotated-sorted-array-ii
三、思路
要尽可能提高效率,可以使用二分查找,但数组中可能有重复的元素,所以需要做改进。升序数组经过旋转之后,必定至少有一边时有序的(以中间元素划分),如果本次查找最左边的元素和中间元素相等,就分不清哪边是升序的,就让左边的元素往走一步,相当于去掉一个重复值。二分查找改进部分:
如果左半边有序,且 target 大于左边元素并小于中间元素,则在左半边继续二分查找,否则就在右半边继续二分查找(并非直接返回false);
如果右半边有序,且 target 大于中间元素并小于右边元素,则在右半边继续二分查找,否则就在左半边继续二分查找(并非直接返回false);
二分查找结束,如果找不到target,则返回false。
四、代码
class Solution {
public:
bool search(vector<int>& nums, int target) {
int len = nums.size();
int l = 0, r = len -1;
while(l <= r){
int mid = l + (r - l) / 2;
if(nums[mid] == target) return true;
if(nums[l] == nums[mid]){ //分不清是哪半边有序,直接去掉一个重复项即可
l++;
continue;
}
if(nums[l] < nums[mid]){ //左半部分升序
if(nums[l] <= target && nums[mid] > target){
r = mid -1;
}else{
l = mid + 1;
}
}
else{ //右半部分升序
if(nums[mid] < target && nums[r] >= target){
l = mid + 1;
}else{
r = mid -1;
}
}
}
return false; //找不到target
}
};