我们把数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/longest-mountain-in-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
先从左到右扫描一次,得到从左到右递增子数组的长度,
再从右往左扫描一次,得到从右往左递增子数组的长度,
最后再从左到右扫描一次,计算以每个点作为山顶的山脉长度即可。
时间复杂度:O(N)
空间复杂度:O(N)
class Solution(object):
def longestMountain(self, A):
"""
:type A: List[int]
:rtype: int
"""
l, r = [0 for _ in A], [0 for _ in A]
for i in range(1, len(A)):
if A[i] > A[i - 1]:
l[i] = l[i - 1] + 1
for i in range(len(A) - 2, -1, -1):
if A[i] > A[i + 1]:
r[i] = r[i + 1] + 1
res = 0
for i in range(len(A)):
if l[i] and r[i] and l[i] + r[i] > 1:
res = max(l[i] + r[i] + 1, res)
return res