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;
}