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;
            }
        }
    }