LintCode : 尋找峯值

LintCode : 尋找峯值

題目

你給出一個整數數組(size爲n)，其具有以下特點：
相鄰位置的數字是不同的
A[0] < A[1] 並且 A[n - 2] > A[n - 1]
假定P是峯值的位置則滿足A[P] > A[P-1]且A[P] > A[P+1]，返回數組中任意一個峯值的位置。

樣例

給出數組[1, 2, 1, 3, 4, 5, 7, 6]返回1, 即數值 2 所在位置, 或者6, 即數值 7 所在位置.

思路

題目中有提到A[0] < A[1] 並且 A[n - 2] > A[n - 1]，並且只用返回數組中任意一個峯值的位置，那麼我們用二分法找到A[mid]，如果A[mid] < A[mid - 1]， 那麼A[0]-A[mid]中必有一個峯值。同理，如果A[mid] < A[mid + 1]，那麼A[mid]-A[n - 1]中必有一個峯值。當然如果A[mid - 1] < A[mid] > A[mid + 1]，那A[mid]就是峯值辣。

代碼

int findPeak(vector<int> A) {
        int p = 0; 
        int r = A.size() - 1;
        while(p < r) {
            int q = (p + r) / 2;
            if(A[q] > A[q - 1] && A[q] > A[q + 1]) {
                    return q;
            }
            else if(A[q] < A[q + 1]) {
                p = q;
            }
            else {
                r = q;
            }
        }
    }