搜索旋转排序数组 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;
    }
};
上一篇 下一篇

猜你喜欢

热点阅读