845. 数组中的最长山脉
我们把数组 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:
def longestMountain(self, A: List[int]) -> int:
# f[i]以i结尾的最长山脉
# g[i]以i结尾的最长递增
f = [0] * len(A)
g = [1] * len(A)
if len(A) <=2:
return 0
if A[0] < A[1]:
g[1] = 2
for i in range(2, len(A)):
if A[i] > A[i-1]:
g[i] = g[i-1] + 1
if A[i] < A[i-1] and f[i-1] == 0 and g[i-1]>1:
f[i] = g[i-1] + 1
if A[i] < A[i-1] and f[i-1] > 0:
f[i] = f[i-1] + 1
return max(f)
方法二:记录所有山脚
遍历数组,记录左右两侧均不小于自身的位置,为山脚,找山脚最大间隔
class Solution:
def longestMountain(self, A: List[int]) -> int:
n = len(A)
ans = left = 0
while left + 2 < n:
right = left + 1
if A[left] < A[left + 1]:
while right + 1 < n and A[right] < A[right + 1]:
right += 1
if right < n - 1 and A[right] > A[right + 1]:
while right + 1 < n and A[right] > A[right + 1]:
right += 1
ans = max(ans, right - left + 1)
else:
right += 1
left = right
return ans