HDU 5654 xiaoxin and his watermelon candy

题意：

给定一个数组，询问区间内有多少个满足要求的不同的三元组，三元组要求： $j = i + 1, k = j + 1$ ,$a_i \leq a_j \leq a_k$

思路：

很裸的主席树求区间内不同的数的个数，这类问题的方法就是，用数组 $pre[i]$ 表示前一个与 $a[i]$ 相同的数的位置，以 $pre$ 数组为关键字，对于询问 $[L, R]$，再用主席树去在区间内查询有多少个数 小于 $L$，因为 $pre[i] \le L - 1$ 说明这个数在 $[L, R]$ 内是不会被重复计数的！

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <map>
#include <cmath>
#include <algorithm>
#include <vector>
using namespace std;

#define N 202000
#define M 400030
#define mod 1000000007
#define LL long long


int n, q;

int a[N];
struct node {
    int x, y, z, p;
    node() {}
    node(int x, int y, int z, int p):x(x), y(y), z(z), p(p) {}
    bool operator < (const node &b) const {
        if(x != b.x) return x < b.x;
        if(y != b.y) return y < b.y;
        if(z != b.z) return z < b.z;
        return p < b.p;
    }
    bool operator == (const node &b) const {
        return x == b.x && y == b.y && z == b.z;
    }
}b[N];
int pre[N];
int rt[N];
int s[N*40], ch[N*40][2], sz;
#define md (ll + rr >> 1)

int update(int i, int l, int r, int v, int ll, int rr) {
    int k = ++sz;
    if(k == N * 40 - 1) {
        vector<int> g; g[1] = -1;
    }
    ch[k][0] = ch[i][0];
    ch[k][1] = ch[i][1];
    s[k] = s[i];
    if(ll == l && rr == r) {
        s[k] += v;
        return k;
    }
    if(r <= md) ch[k][0] = update(ch[i][0], l, r, v, ll, md);
    else if(l > md) {
        ch[k][1] = update(ch[i][1], l, r, v, md + 1, rr);
    }
    else {
        ch[k][0] = update(ch[i][0], l, md, v, ll, md);
        ch[k][1] = update(ch[i][1], md + 1, r, v, md + 1, rr);
    }
    return k;
}
int query(int i, int o, int x, int ll, int rr) {
    int ret = s[i] - s[o];
    if(ll == rr) return ret;
    if(x <= md) ret += query(ch[i][0], ch[o][0], x, ll, md);
    else ret += query(ch[i][1], ch[o][1], x, md + 1, rr);
    return ret;
}

int main() 
{
    int cas;
    scanf("%d", &cas);
    while(cas--) 
    {
        sz = 0;
        scanf("%d", &n);
        for(int i = 1; i <= n; ++i) {
            scanf("%d", &a[i]);
        }
        int cnt = 0;

        for(int i = 1; i <= n - 2; ++i) {
            if(a[i] <= a[i+1] && a[i+1] <= a[i+2]) {
                b[++cnt] = node(a[i], a[i+1], a[i+2], i);
            }
        }
        for(int i = 1; i <= n; ++i) pre[i] = n + 1;
        sort(b + 1, b + cnt + 1);
        for(int i = 1; i <= cnt; ++i) {
            int j = i;
            while(j <= cnt && b[j] == b[i]) ++j;
            --j;
            int x = -1;
            for(int k = i; k <= j; ++k) {
                if(x == -1) pre[b[k].p] = 0;
                else pre[b[k].p] = x;
                x = b[k].p;
            }
            i = j;
        }
        for(int i = 1; i <= n; ++i) {
            rt[i] = rt[i-1];
            if(pre[i] != n + 1) {
                rt[i] = update(rt[i], pre[i] + 1, i, 1, 1, n);
            }
        }
        scanf("%d", &q);
        while(q--) {
            int l, r;
            scanf("%d%d", &l, &r);
            r -= 2;
            if(l > r) {
                puts("0");
                continue;
            }
            printf("%d\n", query(rt[r], rt[l-1], l, 1, n));
        }
    }
    return 0;
}