LeetCode 2951.找出峰值：模拟（遍历）

【LetMeFly】2951.找出峰值：模拟（遍历）

力扣题目链接：https://leetcode.cn/problems/find-the-peaks/

给你一个下标从 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

解题方法：模拟（遍历）

用$i$从$1$到$mountain.length-1$进行枚举，如果$mountain[i]>mountain[i-1]$且$mountain[i]>mountain[i+1]$，就将$i$添加到答案数组中。

小Tips: 将$i$添加到答案数组中的同时，可以将$i++$（因为下一个一定不是“峰值”元素）。

• 时间复杂度$O(mountain.length)$
• 空间复杂度$O(1)$

AC代码

C++

class Solution {
public:
    vector<int> findPeaks(vector<int>& mountain) {
        vector<int> ans;
        for (int i = 1; i < mountain.size() - 1; i++) {
            if (mountain[i] > mountain[i - 1] && mountain[i] > mountain[i + 1]) {
                ans.push_back(i);
                i++;
            }
        }
        return ans;
    }
};
1
2
3
4
5
6
7
8
9
10
11
12
13

Go

// package main

func findPeaks(mountain []int) []int {
    ans := make([]int, 0)
    for i := 1; i < len(mountain) - 1; i++ {
        if mountain[i] > mountain[i - 1] && mountain[i] > mountain[i + 1] {
            ans = append(ans, i)
            i++
        }
    }
    return ans
}
1
2
3
4
5
6
7
8
9
10
11
12

Java

// import java.util.ArrayList;
// import java.util.List;

class Solution {
    public List<Integer> findPeaks(int[] mountain) {
        List<Integer> ans = new ArrayList<>();
        for (int i = 1; i < mountain.length - 1; i++) {
            if (mountain[i] > mountain[i - 1] && mountain[i] > mountain[i + 1]) {
                ans.add(i);
                i++;
            }
        }
        return ans;
    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

Python

# from typing import List

class Solution:
    def findPeaks(self, mountain: List[int]) -> List[int]:
        ans = []
        for i in range(1, len(mountain) - 1):
            if mountain[i] > mountain[i - 1] and mountain[i] > mountain[i + 1]:
                ans.append(i)
        return ans
1
2
3
4
5
6
7
8
9

同步发文于CSDN和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~

Tisfy：https://letmefly.blog.csdn.net/article/details/139279605