您的位置:首页 > 大数据 > 人工智能

POJ 3250 Bad Hair Day(单调栈)

2016-10-13 15:21 323 查看
【题意】:给你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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  单调栈