51nod 1349 最大值（单调栈）

1349 最大值

有一天，小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

输入

第一行读入一个正整数n(1<=n<=100000)，表示有n个数字。
接下来一行读入n个正整数ai(1<=ai<=100000)
接下来一行一个正整数Q(1<=Q<=100000)，表示有Q组询问。
接下来Q行，每行一个正整数k(1<=k<=100000)

输出

Q行，每行一个正整数，表示存在多少区间大于等于k。

输入样例

3
1 2 3
3
1
2
3

输出样例

6
5
3

代码

#include <bits/stdc++.h>

#define ll long long
using namespace std;
const int maxn = 100010;
// r[i]: 以a[i]为最大值可延伸到的区间的右边
// l[i]: 以a[i]为最大值可延伸到的区间的左边
// ans[i]: 以a[i]为最大值的答案数
ll r[maxn], l[maxn], ans[maxn], a[maxn], st[maxn];

int main() {
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
    }
    int top = 0;
    for (int i = 1; i <= n; i++) {
        while (top && a[st[top]] <= a[i]) {
            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]] += 1ll * (i - l[i] + 1) * (r[i] - i + 1);
    }
    for (int i = maxn - 1; i >= 1; i--) {
        ans[i] += ans[i + 1];
    }
    int q;
    cin >> q;
    while (q--) {
        int k;
        cin >> k;
        cout << ans[k] << endl;
    }
    return 0;
}