题意: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;
}