题目: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;
}
};