Bad Hair Day

Bad Hair Day

我的第一道单调栈题目，知道单调栈（点这里）的思想，思路还是比较好想的；

AC_Code:

#include <iostream>
#include <cstdio>
#include <iomanip>
#include <stack>
using namespace std;
typedef long long ll;
const int maxn=8e4+10;

stack<ll>sta,s;
ll num,n;
int main()
{
    scanf("%lld",&n);
    for(ll i=1;i<=n;i++){
        ll h;
        scanf("%lld",&h);
        if( sta.empty()) sta.push(h),s.push(i);
        else{
            while(!sta.empty() ){
                if( h>=sta.top() ){
                    ll f=s.top();s.pop();sta.pop();
                    num+=i-f-1;
                }
                else break;
            }
            sta.push(h);s.push(i);
        }
    }

    sta.pop();s.pop();
    while( !sta.empty() ){
        ll f=s.top();s.pop();sta.pop();
        num+=n-f;
    }
    printf("%lld\n",num);
    return 0;
}