32 lines
747 B
C++
32 lines
747 B
C++
#include<stdcpp.h>
|
|
using namespace std;
|
|
class Solution {
|
|
public:
|
|
vector<int> findPeaks(vector<int>& mountain) {
|
|
vector<int>rlt;
|
|
const int len = mountain.size();
|
|
for(int i = 1 ; i < mountain.size() - 1; i++){
|
|
if(mountain[i] > mountain[i-1] && mountain[i] > mountain[i+1])
|
|
rlt.push_back(i);
|
|
}
|
|
|
|
return rlt;
|
|
}
|
|
};
|
|
int main(){
|
|
Solution sol;
|
|
|
|
vector<int> ex1 = {2,4,4};
|
|
vector<int>rlt1 = sol.findPeaks(ex1);
|
|
for(int i = 0 ; i <rlt1.size();i++)
|
|
cout<<rlt1[i]<<' ';
|
|
cout<<endl;
|
|
|
|
vector<int> ex2 = {1,4,3,8,5};
|
|
vector<int>rlt2 = sol.findPeaks(ex2);
|
|
for(int i = 0 ; i <rlt2.size();i++)
|
|
cout<<rlt2[i]<<' ';
|
|
cout<<endl;
|
|
|
|
return 0;
|
|
} |