搜索旋转排序数组 II
已知存在一个按非降序排列的整数数组 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
二分搜索
本题承接33题 在数组中添加了相等的元素 ,这意味着我们无法直接根据与 nums[0] 的大小关系,将数组划分为两段,即无法通过「二分」来找到旋转点。
所以要先恢复二段性
三叶题解
class Solution {
public:
bool search(vector<int>& nums, int t) {
int n = nums.size();
int l = 0, r = n - 1;
// 恢复二段性
while (l < r && nums[0] == nums[r]) r--;
// 第一次二分,找旋转点
while (l < r) {
int mid = (l + r + 1) >> 1;
if (nums[mid] >= nums[0]) {
l = mid;
} else {
r = mid - 1;
}
}
int idx = n;
if (nums[r] >= nums[0] && r + 1 < n) idx = r + 1;
// 第二次二分,找目标值
int ans = find(nums, 0, idx - 1, t);
if (ans != -1) return true;
ans = find(nums, idx, n - 1, t);
return ans != -1;
}
int find(vector<int>& nums, int l, int r, int t) {
while (l < r) {
int mid = l + r >> 1;
if (nums[mid] >= t) {
r = mid;
} else {
l = mid + 1;
}
}
return nums[r] == t ? r : -1;
}
};
时间复杂度O(n) 如果最坏情况所有数字都是同一个数O(n)
之后的查找旋转点和查找target都是二分 O(logn)
总体时间复杂度时O(n)