题意
n个牛排成一列向右看,牛i能看到牛j的头顶,当且仅当牛j在牛i的右边并且牛i与牛j之间的所有牛均比牛i矮,求所有牛能看到的牛的数量之和
解法
单调栈的一个应用
P5788 【模板】单调栈
洛谷的一个单调栈模板
单调栈,就是一个栈,不过栈内元素保证单调性。即,栈内元素要么从小到大,要么从大到小。而单调栈维护的就是一个数前/后第一个大于/小于他的数。
在这道题中,找到第一个比他高(或等高)的牛,后面的牛也就看不到了
代码
#include<iostream>
#include<cstdio>
#include<stack>
#include<algorithm>
using namespace std;
int main(){
int n;
while(scanf("%d",&n) != EOF){
stack<int>S;
int t;
scanf("%d",&t);
S.push(t);
long long ans=0;
for(int i=1;i<n;i++){
scanf("%d",&t);
while(!S.empty()&&t>=S.top())S.pop();
ans += (long long)S.size();
S.push(t);
}
printf("%lld\n",ans);
}
return 0;
}