pku3250 Bad Hair Day

http://poj.org/problem?id=3250

单调队列，栈模拟

有 n 头牛头朝东站成一列。每头牛有一定的高度，并且能看到其前面高度比它低的牛的头顶，直到被某头高度大于等于它的高度的牛所挡住。计算每头牛能看到的牛头顶的数量之和。

#include <stdio.h>
#include <stack>

using namespace std;

stack<pair<int, int> > s;
long long sum = 0;

void push(int a, int i)
{
    while((!s.empty()) && s.top().first<=a)
    {
        sum += (i - s.top().second - 1);
        s.pop();
    }
    s.push(make_pair(a, i));
}

int main()
{
    int i, n, a;
    scanf("%d", &n);
    for(i=1; i<=n && scanf("%d", &a); i++)
    {
        push(a, i);
    }
    push(1<<31-1, i);
    printf("%lld\n", sum);
    return 0;
}