搜索旋转排序数组 II
2019-08-08 本文已影响0人
王王王王王景
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [0,0,1,2,2,5,6] 可能变为 [2,5,6,0,0,1,2] )。
编写一个函数来判断给定的目标值是否存在于数组中。若存在返回 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
进阶:
这是 搜索旋转排序数组 的延伸题目,本题中的 nums 可能包含重复元素。
这会影响到程序的时间复杂度吗?会有怎样的影响,为什么?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-in-rotated-sorted-array-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
(1)每一次取一个中心点,如果该点就是target,直接返回true;
(2)该点要么左边是有序序列,要么右边是有序序列;
(3)如果左边是有序序列,则判断target是否在这个有序范围内(nums[start] <= target < nums[mid]);
如果在,则进行在[start, mid )的范围内进行搜索;如果不在则取右边非有序序列进行搜索,过程返回到(1)
(4)同样对于右边,如果右边是有序序列,判断target是否属于这个范围之类,如果属于则在该范围内进行搜索,
如果不在则在左边的非有序序列搜索。
代码:
class Solution {
public:
bool search(vector<int>& nums, int target) {
int start = 0, end = nums.size() -1;
while(start <= end) {
// 排除重复数字
while(start < nums.size() - 1 && nums[start] == nums[start + 1])
++start;
while(end > 0 && nums[end] == nums[end - 1])
--end;
int mid = (start + end) / 2;
if(nums[mid] == target) return true;
if(nums[mid] >= nums[start]) {
// 左边是有序的
if(target < nums[mid] && target >= nums[start]) end = mid - 1;
else start = mid + 1;
} else {
// 右边是有序的
if(target > nums[mid] && target <= nums[end]) start = mid + 1;
else end = mid - 1;
}
}
return false;
}
};