[USACO06NOV] Bad Hair Day S

[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 <bits/stdc++.h>
#define int long long

using namespace std;
int n,arr[3000005],ans;

signed main(){
    stack <int> st;
    cin>>n;
    int y=0;
    for(int i=1;i<=n;i++){
        cin>>arr[i];
    }
    for(int i=1;i<=n;i++){
        while(!st.empty()&&arr[i]>=arr[st.top()]){
            y+=i-st.top()-1;
            st.pop();
        }
        st.push(i);
    }
    while(!st.empty()){
        y+=n-st.top();
        st.pop();
    }
    cout<<y;
    return 0;
}