poj3250 Bad Hair Day 单调栈

题意:n个人站成一排,每个人都有身高。对于第i个人,向右找连续比他矮的人的个数记为num[i]。(当h[j]>= h[i]时,i就无

法看到右边比他矮的人),求所有人num的和。

思路:维护一个单调栈,按身高严格降序。从左扫到右,考虑当前的i,设栈顶元素为temp,如果temp.h<p[i].h,那么直接入栈。

不然出栈直到当前栈顶元素的h比p[i].h大,初始一个num=0。因为栈是严格降序的,所以出栈的元素中的人数要算入还在栈中的人

的num中,所以每弹出一个元素,ans+=num,num+=temp.num。最后再将p[i].num+=num,并把p[i]压栈。最后再把栈中元素

依次弹出更新ans即可。详见代码:

// file name: poj3250.cpp //
// author: kereo //
// create time:  2014年11月03日 星期一 13时28分00秒 //
//***********************************//
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<stack>
#include<cmath>
#include<string>
#include<algorithm>
using namespace std;
typedef long long ll;
const int sigma_size=26;
const int MAXN=80000+100;
const double eps=1e-8;
const int inf=0x3fffffff;
const int mod=1000000000+7;
#define L(x) (x<<1)
#define R(x) (x<<1|1)
int n;
struct node{
	ll h,num;
}p[MAXN];
stack<node>s;
int main()
{
	while(~scanf("%d",&n)){
		for(int i=1;i<=n;i++){
			p[i].num=1;
			scanf("%I64d",&p[i].h);
		}
		ll ans=0;
		for(int i=1;i<=n;i++){
			if(s.empty())
				s.push(p[i]);
			else{
				node temp=s.top();
				if(temp.h>p[i].h)
					s.push(p[i]);
				else{
					ll num=0;
					while(!s.empty() && s.top().h<=p[i].h){
						temp=s.top(); s.pop();
						ans+=num;
						num+=temp.num;
					}
					p[i].num+=num;
					s.push(p[i]);
				}
			}
		}
		ll num=0;
		while(!s.empty()){
			node temp=s.top(); s.pop();
			ans+=num;
			num+=temp.num;
		}
		printf("%I64d\n",ans);
	}
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值