有一天,小a给了小b一些数字,让小b帮忙找到其中最大的数,由于小b是一个程序猿,当然写了一个代码很快的解决了这个问题。
这时,邪恶的小c又出现了,他问小b,假如我只需要知道这些数字中的某个区间的最大值,你还能做嘛?
小b经过七七四十九天的思考,终于完美的解决了这道题目,这次,他想也让小c尝尝苦头,于是他问小c,我现在想知道存在多少不同的区间的最大值大于等于k,你还能做吗?
这次,小c犯了难,他来请教身为程序猿的你。
Hint:一个区间指al,al+1,…,ar这一段的数且l<=r,一个区间的最大值指max{al,al+1,…,ar},两个区间不同当且仅当[l1,r1],[l2,r2]中l1不等于l2或r1不等于r2
Input
第一行读入一个正整数n(1<=n<=100000),表示有n个数字。 接下来一行读入n个正整数ai(1<=ai<=100000) 接下来一行一个正整数Q(1<=Q<=100000),表示有Q组询问。 接下来Q行,每行一个正整数k(1<=k<=100000)
Output
Q行,每行一个正整数,表示存在多少区间大于等于k。
Input示例
3 1 2 3 3 1 2 3
Output示例
6 5 3
#include
#include
#include
#define MAXN 100005
using namespace std;
typedef long long LL;
int a[MAXN],l[MAXN],r[MAXN],st[MAXN];
LL Ans[MAXN];
/*
r[i] 表示以a[i] 为最大值的区间的右端点
l[i] 表示以a[i] 为最大值的区间的左端点
Ans[i] 表示以a[i]为区间最大值的区间数
由于是大于等于 还要对ans数组做一个后缀和
*/
inline void read(int &x) {
x = 0; int f = 1; register char c = getchar();
while(c > '9' || c < '0') { if(c == '-') f = -1; c = getchar(); }
while(c >= '0' && c <= '9') { x = x * 10 + c - '0'; c = getchar(); } x *= f;
}
int main(int argc,char *argv[]) {
int n ;read(n);
for(int i=1; i<=n; ++i) read(a[i]);
int top = 0;
for(int i=1; i<=n; ++i) {
while(top && a[i] >= a[st[top]]) top--;
if(top == 0) l[i] = 1;
else l[i] = st[top] + 1;
st[++top] = i;
}
top = 0;
for(int i=n; i>=1; --i) {
while(top && a[st[top]] < a[i]) --top;
if(top == 0) r[i] = n;
else r[i] = st[top] - 1;
st[++top] = i;
}
for(int i=1; i<=n; ++i) Ans[a[i]] += (LL)(i - l[i] + 1) * (r[i] - i + 1);
for(int i=MAXN-1; i>=1; --i) Ans[i] += Ans[i+1];// 大于等于K 而非严格等于K
int q,k; read(q);
while(q--) { read(k); printf("%lld\n",Ans[k]); }
return 0;
}