我们把数组 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
解法一:双指针法
从左向右,先找到左侧山脚i,然后移动j至山峰,接着将j移至右侧山脚。
符合条件的左侧山脚必须满足A[i] < A[i + 1],右侧山峰必须满足A[j] > A[j + 1]。代码中的两处if判断即是此用途。当左侧山脚符合条件时,才会继续后续步骤,否则右移i寻找下一个符合条件的山峰;当右侧山脚符合条件时,才会继续寻找右侧山脚,否则右移i寻找下一个符合条件的山峰。
class Solution:def longestMountain(self, A: List[int]) -> int:ans = i = j = 0n = len(A)while i + 2 < n:j = i + 1# 判断当前i位置是否符合左侧山脚if A[i] < A[i + 1]:# 将j移至山峰while j < n - 1 and A[j] < A[j + 1]:j += 1# 判断当前j位置是否符合右侧山坡起点if j < n - 1 and A[j] > A[j + 1]:# 移动j至右侧山脚while j < n - 1 and A[j] > A[j + 1]:j += 1# 保存最大长度ans = max(ans, j - i + 1)# 如不符合,有两种情况:# 1. 山峰已经是数组最后一个元素,应该返回结果# 2. 山峰出现两个相同的值,应该将i跳至j+1# 第一种情况也可以通过j递增1后赋值给i然后退出循环来实现else:j += 1# 右侧山脚可能是下一个左侧山脚i = jreturn ans
