[Usaco2006 Nov]Bad Hair Day 乱发节

题目描述

输入格式

Line 1: 牛的数量 N。

Lines 2..N+1: 第 i+1 是一个整数，表示第i头牛的高度。

输出格式

Line 1: 一个整数表示c[1] 至 c[N]的和。

由于每头奶牛都可以看到前面第一头比它高的奶牛之前的所有奶牛，也就是说我们用单调栈维护一个单调下降的的序列即可。不用单调队列的原因是奶牛都是先进后出、后进先出。

维护了单调栈之后每次加上单调栈里，每次将答案累加上当前栈中的元素即可。时间复杂度为O(N)。

#include<iostream>
#include<cstring>
#include<cstdio>
#define maxn 80001
using namespace std;
  
inline int read(){
    register int x(0); register char c(getchar());
    while(c<'0'||'9'<c) c=getchar();
    while('0'<=c&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar();
    return x;
}
  
int stack[maxn],a[maxn],top;
int main(){
    int n=read();
    long long ans=0;
    for(register int i=1;i<=n;i++) a[i]=read();
    for(register int i=1;i<=n;i++){
        while(top&&stack[top]<=a[i]) top--;
        stack[++top]=a[i];
        ans+=top-1;
    }
    printf("%lld\n",ans);
    return 0;
}