Leetcode 81.搜索旋转排序数组 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

代码

class Solution {
public:
    bool search(vector<int>& nums, int target) {
        int first = 0;
        int last = nums.size();
        while(first!=last){
            int mid = (first+last)>>1;
            if(nums[mid]==target) return true;
            else if(nums[first]<nums[mid]){
                if(nums[first]<=target&&target<nums[mid])
                    last=mid;
                else
                    first=mid+1;
            }else if(nums[mid]==nums[first]) first++;
            else{
                if(nums[mid]<target&&target<=nums[last-1])
                    first=mid+1;
                else
                    last=mid;
            }
        }
        return false;
    }
};

整体的算法思路与Leetcode 33是一样的。区别主要是两处：

• nums[first]<=nums[mid]时，不代表[first,mid)区域是单调递增的，比如1113111111。但是nums[first]<nums[mid]时可以代表。
• nums[first]==nums[mid]时，该进行怎样的操作。我在这个地方卡了好久。其实只要将first++，跳过相等的元素就可以了。

其余的思路与解法几乎一样，就是二分查找的变种。

结果

参考资源

Leetcode
Leetcode 题解