leetcode_845 数组中的最长山脉

345 阅读1分钟

要求

我们把数组 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])  

image.png

解题思路:我们先将所有可能的山峰点找到,满足山峰点的条件是大于左边的值,小于右边的值,我们将所有找到的山峰值进行遍历,以当前的山峰点为中心,向左或者向右进行贪婪搜索,看看长度是不是能扩张,最终我们在所有的山脉中找到一根最长的。