51nod 1349 最大值（单调栈）

http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1349

题意：

求区间内最大值大于等于k的区间个数。

思路：

利用求出对于以a[i]为最大值的区间范围，pre[i]表示左端范围，aft[i]表示右端范围，则区间个数为$(i-pre[i]+1)*(aft[i]-i+1)$。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector>
#include<stack>
#include<queue>
#include<cmath>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int INF = 0x3f3f3f3f;
const int maxn = 100000+5;

int n;
int a[maxn];
int sta[maxn];
ll ans[maxn];

ll pre[maxn],aft[maxn];

inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

int main()
{
    //freopen("in.txt","r",stdin);
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)  a[i]=read();
        int top = 0;
        for(int i=1;i<=n;i++)
        {
            while(top && a[sta[top]]<=a[i])  top--;
            if(top==0)  pre[i]=1;
            else pre[i]=sta[top]+1;
            sta[++top]=i;
        }
        top=0;
        for(int i=n;i>=1;i--)
        {
            while(top && a[sta[top]]<a[i])  top--;  //这儿特别注意一下
            if(top==0)  aft[i]=n;
            else aft[i]=sta[top]-1;
            sta[++top]=i;
        }
        memset(ans,0,sizeof(ans));
        for(int i=1;i<=n;i++)  ans[a[i]]+=(i-pre[i]+1)*(aft[i]-i+1);
        for(int i=100000;i>=1;i--)  ans[i]+=ans[i+1];
        int q; q=read();
        while(q--)
        {
            int x; x=read();
            printf("%lld
",ans[x]);
        }
    }
    return 0;
}