poj3250 Bad Hair Day

原创于 2015-09-19 11:53:20 发布 · 407 阅读

CC 4.0 BY-SA版权

本文介绍了一种解决“牛视距”问题的方法，通过构建单调递增栈，逐个处理每头牛，以确定其右侧可见牛的数量，并计算总的可见牛数。

题意：一排数字代表一排牛的高度，每头牛只能向右看到比自己矮的牛的头，也看不到比自己高或相等的牛右边的牛头。问一共有多少牛可以被看到。
思路；从右边牛开始，构建一个单调递增栈，遇到比栈顶高的牛，不断出栈，找到它能向右边看到的位置，计数中间能看到牛的个数，再进栈，作为栈顶。

#include<cstdio>
#include<cstring>
#include<stack>
#include<algorithm>
using namespace std;
long long h[80010];
stack<int>s;
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++) scanf("%lld",&h[i]);
        h[n+1]=1<<30;
        while(!s.empty()) s.pop();
        s.push(n+1);
        long long sum=0;
        for(int i=n;i>0;i--)
        {
            int x=s.top();
            while(h[x]<h[i])
            {
                s.pop();
                x=s.top();
            }
            sum+=x-i-1;
            s.push(i);
        }
        printf("%lld\n",sum);
    }
    return 0;
}