leetcode 845.数组中的最长山脉(python)
我们把数组 A 中符合下列属性的任意连续子数组 B 称为 “山脉”:
B.length >= 3
- 存在
0 < i < B.length - 1
使得B[0] < B[1] < ... B[i-1] < B[i] > B[i+1] > ... > B[B.length - 1]
(注意:B 可以是 A 的任意子数组,包括整个数组 A。)
给出一个整数数组 A
,返回最长 “山脉” 的长度。
如果不含有 “山脉” 则返回 0
。
示例 1:
输入:[2,1,4,7,3,2,5]
输出:5
解释:最长的 “山脉” 是 [1,4,7,3,2],长度为 5。
示例 2:
输入:[2,2,2]
输出:0
解释:不含 “山脉”。
提示:
0 <= A.length <= 10000
0 <= A[i] <= 10000
class Solution(object):
def longestMountain(self, A):
if len(A) < 3:
return 0
start, max_peak = 0, 0
reach_peak = False
for i in range(1, len(A) - 1):
if A[i - 1] >= A[i] <= A[i + 1]: # 判断山脚
if reach_peak:
reach_peak = False
max_peak = max(max_peak, i - start + 1)
start = i
elif A[i - 1] < A[i] > A[i + 1]: # 判断山顶
reach_peak = True
if reach_peak:
max_peak = max(max_peak, len(A) - start) # 兼容最后没有山脚的情况
return max_peak