SPOJ DQUERY D-query 主席树

来源：互联网发布：淘宝网跳广场舞的服装编辑：程序博客网时间：2024/04/28 04:50

题目：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;}

0 0