POJ 3250Bad Hair Day(单调栈)

来源:互联网 发布:香港域名注册有限公司 编辑:程序博客网 时间:2024/06/05 05:40

【题意】:给你N((1 ≤ N ≤ 80,000))只牛的高度,每只牛都是向右看,它能看到那些严格低于它的牛的头顶,问所有牛能看到的其他牛的和是多少。
例如 10 3 7 4 12 2这六只牛,第一只能看到 3 7 4这三只牛的头顶,7 能看到 4,12 能看到 2,一共是5;
我们可以找一下,对于第i只牛,有多少牛能看到它。
确定这样的思路,可以读入一只牛的高度,就访问它前面的牛,看看是不是能看到它。这样的话,因为n太大,肯定会超时。
仔细想一下,如果一只牛 i 前面有比它小的牛 j ,那么 牛j 是肯定不能再看到其他的牛的,那么就可以把 牛j 删除掉,再把 牛i 加进去。加入 牛i 的时候前面有多少只牛,就说明有多少只牛能看到它,就在答案里面加入这个值。这样我们就得到了一个单调的序列,删除这只牛的时候可以看做是出栈。

#include<cstdio>#include<stack>using namespace std;int main(){    int n;    while(~scanf("%d",&n)){        long long ans = 0;        stack<long long> a;        for(int i = 1; i <= n; i++){            long long k;            scanf("%lld",&k);            while(!a.empty() && k >= a.top()){                a.pop();            }            a.push(k);            ans += a.size() - 1;        }        printf("%lld\n",ans);    }    return 0;}
0 0
原创粉丝点击