Bad Hair Day S（栈）

P2866 [USACO06NOV] Bad Hair Day S

题目描述

农夫约翰有 $N$ 头奶牛正在过乱头发节。

每一头牛都站在同一排面朝右，它们被从左到右依次编号为 $1,2,\cdots ,N$。编号为 $i$ 的牛身高为 $h_i$。第 $N$ 头牛在最前面，而第 $1$ 头牛在最后面。

对于第 $i$ 头牛前面的第 $j$ 头牛，如果 $h_i>h_{i+1},h_i>h_{i+2},\cdots ,h_i>h_j$，那么认为第 $i$ 头牛可以看到第 $i+1$ 到第 $j$ 头牛。

定义 $C_i$ 为第 $i$ 头牛所能看到的牛的数量。请帮助农夫约翰求出 $C_1+C_2+\cdots +C_N$。

输入格式

输入共 $N+1$ 行。

第一行为一个整数 $N$，代表牛的个数。
接下来 $N$ 行，每行一个整数 $a_i$，分别代表第 $1,2,\cdots ,N$ 头牛的身高。

输出格式

输出共一行一个整数，代表 $C_1+C_2+\cdots +C_N$。

输入输出样例 #1

输入 #1

6
10
3
7
4
12
2

输出 #1

5

说明/提示

数据规模与约定

对于 $100\%$ 的数据，保证 $1\le N\le 8\times 10^4$，$1\le h_i\le 10^9$。

代码内容

#include 
using namespace std;

typedef long long ll;
const ll N=8e4+10;
ll a[N],p[N];

int main()
{
    ll n;
    cin>>n;

    for(ll i=1;i<=n;i++) cin>>a[i];

    ll ans=0,t=0;
    for(ll i=1;i<=n;i++)
    {
        while(t&&a[i]>=p[t]) t--;
        ans+=t;
        p[++t]=a[i];
    }

    cout<<ans<<endl;
    return 0;
}