LeetCode 81. 搜索旋转排序数组 II（Search in Rotated Sorted Array II）

题目描述

 

假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如，数组 [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  可能包含重复元素。
• 这会影响到程序的时间复杂度吗？会有怎样的影响，为什么？

 

解题思路

 

由于本题中的nums可能包含重复元素，所以在进行二分查找时要多考虑一种情况，即当left、mid和right指向的数字都相等时，是无法判断出目标数在哪个半区里，所以只能线性查找。

 

代码

 

class Solution {
public:
    bool search(vector<int>& nums, int target) {
        int left = 0, mid, right = nums.size() - 1;
        while(left <= right){
            mid = (left + right) / 2;
            if(nums[mid] == target) return true;
            if(nums[left] == nums[mid] && nums[right] == nums[mid]){
                for(; left < right; left++)
                    if(nums[left] == target) return true;
                return false;
            }
            if(nums[mid] < target){
                if(nums[right] >= target || nums[mid] >= nums[left])
                    left = mid + 1;
                else right = mid - 1;
            }
            else{
                if(nums[left] <= target || nums[mid] <= nums[right])
                    right = mid - 1;
                else left = mid + 1;
            }
        }
        return false;
    }
};