假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [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
这是 搜索旋转排序数组 的延伸题目,本题中的 nums 可能包含重复元素。
这会影响到程序的时间复杂度吗?会有怎样的影响,为什么?
主要是分情况,left
到right
之间分为有断裂的情况和无断裂的情况。前者又分为mid
在左半分支还是右半分支,后者直接二分。另外需要注意相等的情况,可能需要同时移动left
和right
。
注意考虑到所有情况,否则可能一直在循环里边。
运行时间0ms,代码如下。
bool search(int* nums, int numsSize, int target) {
int left = 0, right = numsSize - 1, mid = 0;
while(left <= right) {
mid = (left + right) >> 1;
if(nums[mid] == target)
return true;
if(nums[left] > nums[right] && nums[mid] >= nums[left]) {
if(nums[left] <= target && target < nums[mid])
right = mid - 1;
else
left = mid + 1;
}
else if(nums[left] > nums[right] && nums[mid] <= nums[right]) {
if(nums[mid] < target && target <= nums[right])
left = mid + 1;
else
right = mid - 1;
}
else if(nums[left] < nums[right]) {
if(nums[mid] > target)
right = mid - 1;
else
left = mid + 1;
}
else if(nums[left] == nums[right]) {
if(nums[left] != target) {
left++;
right--;
}
else
return true;
}
}
return false;
}