1. 程式人生 > 其它 >C++ IMPL模式解析(下)

C++ IMPL模式解析(下)

已知存在一個按非降序排列的整數陣列 nums ,陣列中的值不必互不相同。

在傳遞給函式之前,nums 在預先未知的某個下標 k0 <= k < nums.length)上進行了 旋轉 ,使陣列變為 [nums[k], nums[k+1], ..., nums[n-1], nums[0], nums[1], ..., nums[k-1]](下標 從 0 開始 計數)。例如, [0,1,2,4,4,4,5,6,6,7] 在下標 5 處經旋轉後可能變為 [4,5,6,6,7,0,1,2,4,4] 。

給你 旋轉後 的陣列 nums 和一個整數 target ,請你編寫一個函式來判斷給定的目標值是否存在於陣列中。如果 nums

 中存在這個目標值 target ,則返回 true ,否則返回 false 。

你必須儘可能減少整個操作步驟。

class Solution {
public:
    bool search(vector<int>& a, int x) {
        int R = a.size() - 1;
        while (R >= 0 && a[R] == a[0]) R--;
        if (R < 0) return a[0] == x;

        int l = 0, r = R;
        while (l < r) {
            int mid = l + r + 1 >> 1;
            if (a[mid] >= a[0]) l = mid;
            else r = mid - 1;
        }

        if (x >= a[0]) l = 0;
        else {
            l = r + 1;
            r = R;
        }

        while (l < r) {
            int mid = l + r + 1 >> 1;
            if (a[mid] <= x) l = mid;
            else r = mid - 1;
        }
        if (a[r] == x) return true;
        return false;
    }
};