要求
我们把数组 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
解释:不含 “山脉”。
核心代码
class Solution:
def longestMountain(self, arr: List[int]) -> int:
if len(arr) < 3:
return 0
def is_peak(index):
return 1 <= index <=len(arr) - 2 and arr[index - 1] < arr[index] and arr[index] > arr[index + 1]
def find_all_peaks():
return [index for index in range(len(arr)) if is_peak(index)]
def get_len_of_the_moutain(index):
left = right = index
length = 1
while left > 0 and arr[left - 1] < arr[left]:
left -= 1
length += 1
while right < len(arr) - 1 and arr[right] > arr[right + 1]:
right += 1
length += 1
return length
return max([get_len_of_the_moutain(peak) for peak in find_all_peaks()] + [0])
解题思路:我们先将所有可能的山峰点找到,满足山峰点的条件是大于左边的值,小于右边的值,我们将所有找到的山峰值进行遍历,以当前的山峰点为中心,向左或者向右进行贪婪搜索,看看长度是不是能扩张,最终我们在所有的山脉中找到一根最长的。