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 = +;

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

 ll pre[maxn],aft[maxn];

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

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