题解 | 打印极值点下标

#include <iostream>
#include <vector>
using namespace std;

int main() {
    int n;
    while(cin>>n){
        vector<int> extremas;
        vector<int> nums;
        for(int i=0;i<n;i++){
            int num;
            cin>>num;
            nums.push_back(num);
        }
        if(nums[1]!=nums[0]) extremas.push_back(0);
        for(int i=1;i<n-1;i++){
            if((nums[i]>nums[i-1]&&nums[i]>nums[i+1])||(nums[i]<nums[i-1]&&nums[i]<nums[i+1])){
                extremas.push_back(i);
            }
        }
        if(nums[n-1]!=nums[n-2]) extremas.push_back(n-1);
        for(int i:extremas){
            cout<<i<<" ";
        }
        cout<<endl;
    }
    return 0;
}