BOJ/백준 3015 오아시스 재결합

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

#define ll long long
#define height first
#define count second 


int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);

    stack<pair<int, int>> s;
    ll ans = 0;

    int n;
    cin >> n;

    vector<int> h(n);
    for(int i = 0; i < n; i++){
        cin >> h[i];
    }

    for(int i = 0; i < n; i++){
        int tmp_h = h[i];
        int cnt = 1;
        while(!s.empty() && s.top().height <= h[i]){
            ans += s.top().count;
            if (s.top().height == tmp_h){
                cnt += s.top().count;
            }
            s.pop();
        }
        if(!s.empty()){
            ans += 1;
        }
        s.push({tmp_h, cnt});
    }
    cout << ans << '\n';
    return 0;
}