백준 3015 오아시스 재결합

https://www.acmicpc.net/problem/3015

키가 같은 경우에서 애 먹었습니다.

#include <iostream>
#include <stack>
using namespace std;
 
int main(){
    ios_base::sync_with_stdio(false);    //cin 속도 증가시키기 위해
    stack<pair<int,int>> st;            //first는 키, second는 키가 같은 경우
    int N,tmp, chk;                     //tmp는 키, chk는 키가 같은 경우
    long long count = 0;
    cin >> N;
    
    for(int i = 0; i < N; i++){
        cin >> tmp;
        chk = 1;
        while(!st.empty() && st.top().first <= tmp){
                count += st.top().second;
                if(st.top().first == tmp) chk+= st.top().second;
                st.pop();
        }
        if(!st.empty()) count +=1;;
        
        st.push({tmp,chk});    
    }
    cout << count;
}