Suppose an array sorted in ascending order is rotated at some pivot unknown to you beforehand.
(i.e., [0,0,1,2,2,5,6]
might become [2,5,6,0,0,1,2]
).
You are given a target value to search. If found in the array return true
, otherwise return false
.
Example 1:
Input: nums = [2,5,6,0,0,1,2]
, target = 0
Output: true
Example 2:
Input: nums = [2,5,6,0,0,1,2]
, target = 3
Output: false
Follow up:
- This is a follow up problem to Search in Rotated Sorted Array, where
nums
may contain duplicates. - Would this affect the run-time complexity? How and why?
class Solution {
public:
bool search(vector<int>& nums, int target) {
int len = nums.size();
unordered_set<int> s;
vector<int> v;
for(int num : nums){
if(!s.count(num)){
s.insert(num);
v.push_back(num);
}
}
int l, m, r;
l = 0; r = v.size() - 1;
while(l <= r){
m = (l + r) / 2;
if(target == v[m])
return true;
else if(v[m] <= v[r]){
if(v[m] < target && target < v[r])
l = m + 1;
else if(target == v[r])
return true;
else
r = m - 1;
} else if(v[l] <= v[m]){
if(v[l] < target && target < v[m])
r = m - 1;
else if(target == v[l])
return true;
else
l = m + 1;
}
}
return false;
}
};