题目来源: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;
}