题目:
你给出一个整数数组(size为n),其具有以下特点:
- 相邻位置的数字是不同的
- A[0] < A[1] 并且 A[n - 2] > A[n - 1]
假定P是峰值的位置则满足A[P] > A[P-1]
且A[P] > A[P+1]
,返回数组中任意一个峰值的位置。
Yes
样例
给出数组[1, 2, 1, 3, 4, 5, 7, 6]
返回1
, 即数值 2 所在位置, 或者6
, 即数值 7 所在位置.
注意
解答:直接查找太简单,时间复杂度为O(n)。可用二分查找,时间复杂度为O(lgn)
数组可能包含多个峰值,只需找到其中的任何一个即可
代码如下:
public int findPeak(int[] A) {
int begin = 0;
int end = A.length - 1;
while(begin < end) {
int mid = begin + (end - begin)/2;
if(mid == 0)
return 1;
if(mid == A.length - 1)
return mid - 1;
if(A[mid] > A[mid - 1] && A[mid] > A[mid + 1])
return mid;
else if(A[mid] < A[mid - 1])
end = mid - 1;
else
begin = mid + 1;
}
return begin;
}