题目描述
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [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
算法1
(线性搜索 + 二分) $O(n)$
Leetcode 33的follow up,因为存在重复元素,旋转处两边可能是相同元素,即数组头元素、尾元素可能相同,从尾部线搜索找到不同于首元素的位置,在数组头元素和该元素之间寻找target即可。
时间复杂度
因为存在线搜索步骤,最坏情况时间复杂度为$O(n)$
参考文献
C++ 代码
class Solution {
public:
bool search(vector<int>& nums, int target) {
int n = nums.size(), l = 0, r = n - 1;
while (r > 0 && nums[r] == nums[0]) --r;
while (l <= r){
int mid = l + r >> 1;
if (target >= nums[0]){
if (nums[mid] < nums[0] || nums[mid] >= target) r = mid - 1;
else l = mid + 1;
}
else {
if (nums[mid] >= nums[0] || nums[mid] < target) l = mid + 1;
else r = mid - 1;
}
}
if (l < n && nums[l] == target) return true;
return false;
}
};