POJ - 3250 G - Bad Hair Day

https://vjudge.net/contest/177579#problem/G


这道题看起来可能有一点难，但是你把题目转换一下就好做了，把这个题求一头牛能被多少头牛看见就行了，用单调栈来做


#include<cstdio>
#include<stack>//用栈做
#include<string>
#include<iostream>
using namespace std;
int main(){
    int n;
    int a[100010];
    long long ans=0;//最坏情况会超出int范围
    stack<int>s;//定义一个栈
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    for(int i=1;i<=n;i++){
        while(!s.empty()&&s.top()<=a[i])//单调栈
        s.pop();
        ans+=s.size();
        s.push(a[i]);
    }
    printf("%I64d\n",ans);
}