你给出一个整数数组(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 所在位置.
Python:
# method1: for loop 遍历
class Solution:
#@param A: An integers list.
#@return: return any of peek positions.
def findPeak(self, A):
if A is None:
return None
elif len(A) < 3:
return 0
for i in range(1, len(A) - 1):
if A[i] > A[i - 1] and A[i] > A[i + 1]:
return i
# method2: 二分
class Solution:
#@param A: An integers list.
#@return: return any of peek positions.
def findPeak(self, A):
if A is None:
return None
elif len(A) < 3:
return 0
left, right = 1, len(A) - 2
while left + 1 < right:
mid = (left + right) / 2
if A[mid] < A[mid - 1]:
right = mid
elif A[mid] < A[mid + 1]:
left = mid
else:
return mid
if A[left] < A[right]:
return right
else:
return left
想法:题设已知A[0] < A[1] 并且 A[n - 2] > A[n - 1],
若A[mid] < A[mid - 1],则左侧必定有一个峰值,否则A[0] > A[1] > … > A[mid-1] > A[mid]成立,与题设不符,因此mid左侧必有峰值;反之,若A[mid] < A[mid + 1],右侧必有峰值。