LeetCode
找出峰值
题目链接:2951. 找出峰值 - 力扣(LeetCode)
题目描述
给你一个下标从 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 <= 1001 <= mountain[i] <= 100
思路
遍历下标在 [ 1, n - 2 ]内的所有数,如果其大于左右两侧相邻数字,则把 i 加入答案。
代码
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);
}
}
return ans;
}
};
Java
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);
}
}
return ans;
}
}