POJ 3250 Bad Hair Day （栈的简单应用）

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

题意：一排全都向右看齐的牛，第 i 头牛可以看到在它右边且比它低的不被比它高的牛所遮挡的牛。问1~n头能看见的和。

刚开始我用最基础的方法写，但是超时了，当然poj也不会出这样简单的水题.太天真了。

超时代码。

#include<cstdio>

__int64 a[80005],ans;

int main()
{
    int n;
    scanf("%d",&n);
    for(int i = 0;i < n; i++)
        scanf("%I64d",&a[i]);

    ans = 0;

    for(int i = 0;i < n; i++){
        for(int j = i + 1;j < n; j++){

            if(a[i] > a[j])
                ans++;
            else break;

        }
    }

    printf("%I64d\n",ans);

    return 0;

}

后来发现可以用栈的数组模拟来写。

思路：只要每次把比输入的一头牛低的或者等于它的牛去掉，那么栈中剩余的牛便是能看见这头牛的数量。

#include<cstdio>

__int64 Stack[80005],ans,top;

int main()
{
 //   freopen("in.txt","r",stdin);
    int n;
    scanf("%d",&n);

    ans = 0;
    top = 0;

    for(int i = 0;i < n; i++){
        __int64 num;
        scanf("%I64d",&num);

        while(top > 0 && Stack[top-1] <= num) top--;

        ans += top;

        Stack[top++] = num;

    }

    printf("%I64d\n",ans);

    return 0;

}