3.8 Find Peak Element寻找峰值
在线题测:http://www.lintcode.com/en/problem/find-peak-element/
在线题解:https://www.jiuzhang.com/solution/find-peak-element/
75.寻找峰值
你给出一个整数数组(size为n),其具有以下特点:
- 相邻位置的数字是不同的
- A[0] < A[1] 并且 A[n - 2] > A[n - 1]
假定P是峰值的位置则满足A[P] > A[P-1]且A[P] > A[P+1],返回数组中任意一个峰值的位置。
样例
样例 1:
输入: [1, 2, 1, 3, 4, 5, 7, 6]
输出: 1 or 6
解释:
返回峰顶元素的下标
样例 2:
输入: [1,2,3,4,1]
输出: 3
挑战
Time complexity O(logN)
注意事项
- 数组保证至少存在一个峰
- 如果数组存在多个峰,返回其中任意一个就行
- 数组至少包含 3 个数
更新代码
class Solution:
#@param A: An integers list.
#@return: return any of peek positions.
def findPeak(self, A):
# write your code here
start, end = 1, len(A) - 2
while start + 1 < end:
mid = (start + end) // 2
if A[mid] < A[mid - 1]:
end = mid
elif A[mid] < A[mid + 1]:
start = mid
else:
end = mid
if A[start] < A[end]:
return end
else:
return start