有一天,小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 <iostream>
using namespace std;
typedef long long int ll;
const int MAXN = 100005;
int n, k, input[MAXN], q[MAXN];
ll ans[MAXN], sum[MAXN];
int main()
{
cin >> n;
for (int i = 1;i <= n; i++)
{
cin >> input[i];
}
int r = 0;
input[n + 1] = MAXN;
for (int i = 1; i <= n + 1; i++)
{
while ((input[i] >= input[q[r]]) && (r >= 1))
{
ans[input[q[r]]] += (ll)(q[r] - q[r - 1]) * (i - q[r]);
r--;
}
q[++r] = i;
}
for (int i = 100000; i > 0; i--)
{
sum[i] = sum[i + 1] + ans[i];
}
int num;
cin >> num;
while (num--)
{
cin >> k;
cout << sum[k] << endl;
}
return 0;
}