SPOJ D-query（主席树入门题 求区间不同数的个数）

题目链接

主席树入门题，感觉很久没碰数据结构了，回来再做一遍。这题也能用普通莫队做。
主席树也就是持久化线段树，简单来说就是在每个位置都建一棵线段树，但如果重新建一棵完整的线段树肯定TLE，主席树就利用了更新一个点只跟一条链有关的性质，每次除了新建一条链，其他的点都是拿上一个位置的直接用。
这道题主席树底部的节点表示的是p位置上的数（离散化后）是不是一个不同的数，就更新操作来讲，如果前面位置lst也是这个数，那就把多建一条把跟lst相关的点都-1的链，然后再建一条跟当前位置p相关的点都+1的链，其余的点都是直接拿上一个数组的来用。
查询l，r区间的不同的数，只要在第 r 棵“线段树”找大于 l 的区间里有多少个不同的数。

#include <bits/stdc++.h>

using namespace std;

const int maxn = 3*(int)1e4+100;
int n, a[maxn], b[maxn], t[maxn*40], lt[maxn*40], rt[maxn*40], tVal[maxn*40], tot;
int lst[maxn];    //记录离散化后数组的上次出现位置
int q;

void init() {
    tot = 0;
    memset(lst, -1, sizeof(lst));
}
int build(int l, int r) {
    int node = ++tot;
    tVal[node] = 0;
    int mid = (l + r) >> 1;
    if (l < r) {
        lt[node] = build(l, mid);
        rt[node] = build(mid+1, r);
    }
    return node;
}
int update(int l, int r, int par, int p, int val) {
    int node = ++tot;
    lt[node] = lt[par]; rt[node] = rt[par]; tVal[node] = tVal[par] + val;
    int mid = (l + r) >> 1;
    if (l < r) {
        if (p <= mid) lt[node] = update(l, mid, lt[par], p, val);
        else rt[node] = update(mid+1, r, rt[par], p, val);
    }
    return node;
}
int query(int l, int r, int p, int node) {
    if (l == r) return tVal[node];
    int mid = (l + r) >> 1;
    if (p <= mid) return (query(l, mid, p, lt[node]) + tVal[rt[node]]);
    else return query(mid+1, r, p, rt[node]);
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.precision(10);
    cout << fixed;
#ifdef LOCAL_DEFINE
    freopen("input.txt", "r", stdin);
#endif

    init();
    cin >> n;
    for (int i = 1; i <= n; ++i) {
        cin >> a[i];
        b[i] = a[i];
    }
    sort(b + 1, b + 1 + n);
    int len = unique(b + 1, b + 1 + n) - b;
    for (int i = 1; i <= n; ++i)
        a[i] = lower_bound(b + 1, b + 1 + len, a[i]) - b;

    t[0] = build(1, n);
    for (int i = 1; i <= n; ++i) {
        if (lst[a[i]] == -1) t[i] = update(1, n, t[i-1], i, 1);
        else {
            int temp = update(1, n, t[i-1], lst[a[i]], -1);
            t[i] = update(1, n, temp, i, 1);
        }
        lst[a[i]] = i;
    }

    cin >> q;
    while (q--) {
        int l, r;
        cin >> l >> r;
        cout << query(1, n, l, t[r]) << '\n';
    }

#ifdef LOCAL_DEFINE
    cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
#endif
    return 0;
}