LeetCode 81-Search in Rotated Sorted Array II
题干:
给定一个不降序数组 n u m s nums nums(里面可以有相同元素),现将 n u m s nums nums从 k k k位置( k k k未知)旋转(即将从 k k k开始到末尾的部分移到最前面),问元素 t a r g e t target target是否在 n u m s nums nums中,如果在返回true,否则返回false。
Input: nums = [2,5,6,0,0,1,2], target = 0
Output: true
解:
画图:
因为有了相等元素,所以旋转数组不满足二段性了(意味着不能使用二分了),所以需要想办法恢复二段性。
若能排除最前面一段与最后面一段相等的那部分,数组就有了二段性,那么就和LeetCode 33-Search in Rotated Sorted Array题一样了,可以用相同的方法解决。
那么怎么排除最前面一段和最后面一段相等的部分呢(其实在LeetCode 154-Find Minimum in Rotated Sorted Array II这个题中给出了解决思路)?在二分循环开始之前,先判断
n
u
m
s
[
l
e
f
t
]
nums[left]
nums[left]是否等于
n
u
m
s
[
r
i
g
h
t
]
nums[right]
nums[right],如果成立,说明是最前面最后面相等的段,可以通过++left
或--right
(根据实际情况选择其一)去除重复元素。
bool search(vector<int>& nums, int target){
int left = 0, right = nums.size()-1;
bool flag = 0;
while (left < right && nums[left] == nums[right]) //去掉最前面段的重复数字
--right;
while(left <= right){
int mid = left + ((right - left) >> 1);
if(nums[mid] == target){
flag = 1;
break;
}
if(nums[mid] >= nums[0]){ // mid在分界点左边,[left,mid]这部分是有序的
if(target >= nums[left] && target < nums[mid])
right = mid - 1;
else left = mid + 1;
}
else{ //mid在分界点右边,[mid, right]这部分是有序的
if(target > nums[mid] && target <= nums[right])
left = mid + 1;
else right = mid - 1;
}
}
return flag;
}