找出峰值

标签: 数组 枚举

难度: Easy

给你一个下标从 0 开始的数组 mountain 。你的任务是找出数组 mountain 中的所有 峰值

以数组形式返回给定数组中 峰值 的下标,顺序不限

注意:

  • 峰值 是指一个严格大于其相邻元素的元素。
  • 数组的第一个和最后一个元素 是峰值。

示例 1:

输入:mountain = [2,4,4]
输出:[]
解释:mountain[0] 和 mountain[2] 不可能是峰值,因为它们是数组的第一个和最后一个元素。
mountain[1] 也不可能是峰值,因为它不严格大于 mountain[2] 。
因此,答案为 [] 。

示例 2:

输入:mountain = [1,4,3,8,5]
输出:[1,3]
解释:mountain[0] 和 mountain[4] 不可能是峰值,因为它们是数组的第一个和最后一个元素。
mountain[2] 也不可能是峰值,因为它不严格大于 mountain[3] 和 mountain[1] 。
但是 mountain[1] 和 mountain[3] 严格大于它们的相邻元素。
因此,答案是 [1,3] 。

提示:

  • 3 <= mountain.length <= 100
  • 1 <= mountain[i] <= 100

Submission

运行时间: 23 ms

内存: 16.0 MB

class Solution:
    def findPeaks(self, mountain: List[int]) -> List[int]:
        res = []
        for i in range(1, len(mountain) - 1):
            if mountain[i] > mountain[i - 1] and mountain[i] > mountain[i + 1]:
                res.append(i)
        return res

Explain

此题解的思路是遍历数组mountain,从第二个元素到倒数第二个元素(因为第一个和最后一个元素不可能是峰值)。对于每个元素,检查它是否严格大于它的左右相邻元素。如果是,那么这个元素就是一个峰值,将其下标添加到结果列表中。

时间复杂度: O(n)

空间复杂度: O(n)

class Solution:
    def findPeaks(self, mountain: List[int]) -> List[int]:
        res = []  # 结果列表
        for i in range(1, len(mountain) - 1):  # 遍历数组,跳过第一个和最后一个元素
            if mountain[i] > mountain[i - 1] and mountain[i] > mountain[i + 1]:  # 检查是否为峰值
                res.append(i)  # 将峰值的下标添加到结果列表中
        return res  # 返回结果列表

Explore

在寻找峰值的过程中,排除数组的第一个和最后一个元素是因为这两个位置没有两边的邻居元素,因此无法判断它们是否大于其相邻的元素。如果数组在边界处峰值,例如数组是单调递增或递减的,那么按照这种方法确实会错过边界峰值。例如,在一个递增的数组[1, 2, 3]中,元素3是一个峰值,但因为它是数组的最后一个元素,我们的方法不会将其识别为峰值。

这种方法不能正确处理所有连续重复元素的情况。例如在数组[1, 2, 2, 3, 2]中,按照题解的方法,数字2和2之间并没有形成峰值,因为它们不是严格大于其左右相邻的元素。尽管实际上数字3是一个峰值,但如果数组中有连续的重复数字,则这种方法可能无法正确识别所有情况的峰值。

如果数组`mountain`的所有元素完全相同,题解的逻辑是正确的,因为没有任何一个元素严格大于其左右相邻的元素。因此,这种情况下,题解会正确地判断出数组中没有峰值。