POJ 3250 单调栈
一群高度不完全相同的牛从左到右站成一排,每头牛只能看见它右边的比它矮的牛的发型,若遇到一头高度大于或等于它的牛,则无法继续看到这头牛后面的其他牛。
给出这些牛的高度,要求每头牛可以看到的牛的数量的和。
把要求作一下转换,其实就是要求每头牛被看到的次数之和。这个可以使用单调栈来解决。
从左到右依次读取当前牛的高度,从栈顶开始把高度小于或等于当前牛的高度的那些元素删除,此时栈中剩下的元素的数量就是可以看见当前牛的其他牛的数量。把这个数量加在一起,就可以得到最后的答案了。
(转自blog.csdn.net/alongela/article/details/8229702)
#include <stack>
#include <cstdio>
using namespace std;
stack<int> st;
int main(){
long long n,ans=0;
scanf("%lld",&n);
for(int i=1;i<=n;i++){
int jy=1,xx;
scanf("%d",&xx);
while(st.size()&&st.top()<=xx)st.pop();
ans+=st.size();
st.push(xx);
}
printf("%lld\n",ans);
}