SPOJ DQUERY D-query 主席树

题目：http://www.spoj.com/problems/DQUERY/en/

题意：给定一个数组，每次询问一个区间内不同元素的个数

思路：看的别人的思路。对于一个数,如果以前没出现过就插入到主席树中否则就删除以前那个。再插入主席树。注意，所有的更新和删除都是建立了新的节点来保持其历史状态的。。对于T[i]我们存的是从1到i区间的不同的数出现了多少个。然后这棵树是根据T[i - 1]来建立的。每个节点表示的是所管辖区间内出现的不同元素个数

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

const int N = 30010;
int n, m;
int rson[N*20], lson[N*20], sum[N*20], root[N];
int arr[N], brr[N], tot;
int vis[N];
int res;
void update(int pre, int &rt, int l, int r, int k, int v)
{
    rt = ++tot;
    lson[rt] = lson[pre], rson[rt] = rson[pre], sum[rt] = sum[pre] + v;
    if(l == r) return;
    int mid = (l + r) >> 1;
    if(k <= mid) update(lson[pre], lson[rt], l, mid, k, v);
    else update(rson[pre], rson[rt], mid + 1, r, k, v);
}
void query(int rt, int l, int r, int k)
{
    if(k == l)
    {
        res += sum[rt];
        return;
    }
    int mid = (l + r) >> 1;
    if(k <= mid) res += sum[rson[rt]], query(lson[rt], l, mid, k);
    else query(rson[rt], mid + 1, r, k);
}
int main()
{
    scanf("%d", &n);
    tot = 0;
    for(int i = 1; i <= n; i++) scanf("%d", &arr[i]), brr[i] = arr[i];
    sort(brr + 1, brr + 1 + n);
    int len = unique(brr + 1, brr + 1 + n) - brr - 1;
    memset(vis, -1, sizeof vis);
    int tmp;
    for(int i = 1; i <= n; i++)
    {
        arr[i] = lower_bound(brr + 1, brr + 1 + len, arr[i]) - brr;
        if(vis[arr[i]] == -1) update(root[i-1], root[i], 1, n, i, 1);//之前不存在当前元素，直接插入树中
        else update(root[i-1], tmp, 1, n, vis[arr[i]], -1), update(tmp, root[i], 1, n, i, 1); //当前元素在树中以存在，用新建节点的方式删除之前的，再插入
        vis[arr[i]] = i;
    }
    scanf("%d", &m);
    int a, b;
    while(m--)
    {
        scanf("%d%d", &a, &b);
        res = 0;
        query(root[b], 1, n, a);
        printf("%d\n", res);
    }
    return 0;
}