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 題解