单调栈的运用

题目来源：poj3250 题目链接

题目大意：有一群列队，面朝右站立的牛，输入每头牛的身高，如果视线前方有一头高于自己的牛，则看不到它之后的牛；求出所有牛能看到头顶的个数。

n方算法超时，故考虑优化，首先能想到的是，在计算左边的牛能看到多少牛时，可以利用右边的结果。栈优化的思路即是，从右向左遍历牛的身高，同时对栈进行操作（栈中存放牛的编号），栈首牛低于当前牛，则去掉该元素；反之则说明该牛最远能看到的，是栈首牛的前一头牛。如果栈空，则说明能一直看到头。至于为何该算法是正确的，核心原因在于，靠前的牛的身高对看的距离远近影响至关重要，所以用靠前的牛的身高来更新栈是正确的。即每当出现较高的牛，栈中后面那些身高较低的牛可以被忽略，一直忽略到比这个牛更高的牛为止，因为如果连这头牛的身高都无法达到，就没有必要再其后的矮牛设卡；而在身后更高的牛处设卡，就表示越过这个障碍时能一直看到那么远。

AC代码：

 

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;

int st[80010],h[80010];
int n,p;
long long ans;

int main()
{
    while(cin>>n)
    {
        for(int i=1; i<=n; i++)
            scanf("%d",h+i);
        p=-1,ans=0;
        for(int i=n; i>0; i--)
        {
            while(p>=0&&h[i]>h[st[p]])
                p--;
            ans+=p==-1?n-i:st[p]-i-1;
            st[++p]=i;
        }
        cout<<ans<<endl;
    }
    return 0;
}