题目描述
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [0,0,1,2,2,5,6] 可能变为 [2,5,6,0,0,1,2] )。
编写一个函数来判断给定的目标值是否存在于数组中。若存在返回 true,否则返回 false。
示例:
输入: nums = [2,5,6,0,0,1,2], target = 0
输出: true
输入: nums = [2,5,6,0,0,1,2], target = 3
输出: false
题目链接: https://leetcode-cn.com/problems/search-in-rotated-sorted-array-ii/
思路
这题是搜索旋转排序数组的延伸版。在“搜索旋转数组”那题中,首先通过num[left],nums[right]和nums[mid]之间的关系来判断接下来搜索的范围。在这题中,数组中可能存在重复元素,也就是可能存在nums[left]==nums[mid]==nums[right]的情况,例如[1,3,1,1,1]、[1,1,1,3,1]。这种情况出现时,我们没有办法缩小搜索的范围,只能在[left, right]范围内顺序查找。所以,代码就是在搜索旋转排序数组这题代码的基础上加了一个判断条件,具体如下:
class Solution {
public:
bool search(vector<int>& nums, int target) {
if(nums.empty()) return false;
int left = 0, right = nums.size()-1;
while(left<=right){
int mid = left+(right-left)/2;
if(nums[mid]==target) return true;
if(nums[left]==nums[mid] && nums[mid]==nums[right]){ // 三者相等,遍历寻找
for(int i=left; i<=right; i++){
if(nums[i]==target) return true;
}
return false;
}
if(nums[left]<=nums[mid]){
if(nums[left]<=target && target<nums[mid]) right = mid-1;
else left = mid+1;
}else{ // 也可以写成else if(nums[mid]<=nums[right]),更易读
if(nums[mid]<target && target<=nums[right]) left = mid+1;
else right = mid-1;
}
}
return false;
}
};
- 时间复杂度:O(n)+O(logn)
这个时间复杂度不一定准确。 - 空间复杂度:O(1)