LintCode 75 寻找峰值

题目:findPeak

---

要求:

你给出一个整数数组(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 所在位置.

算法要求:

Time complexity O(logN)

解题思路:

用二分法即可。

算法如下:

class Solution {public:    /**     * @param A: An integers array.     * @return: return any of peek positions.     */    int findPeak(vector<int> A) {        // write your code here        int left = 0;          int size = A.size();        int right = size - 1;          while(left < right) {              int mid = left + (right - left)/2;              if(mid == 0) {                return 1;             } else if(mid == size - 1) {                return mid - 1;              } else if(A[mid] > A[mid - 1] && A[mid] > A[mid + 1]) {                 return mid;              } else if(A[mid] < A[mid - 1]) {                right = mid - 1;              } else {                  left = mid + 1;              }        }          return left;      }};