spoj-D-query(主席树)

题目链接

1.思路

扫描数组，以root[i]为根的线段树上记录的是从第一个位置到i位置的数字信息，即当前线段树上的节点记录的是数组的位置信息；

2.code

#include <cstdio>
#include <map>
using namespace std;
const int N = 3e4 + 5;

struct segTree {
    int lc, rc, cnt;
};

segTree ret[N*20];
int root[N];
int o;

void build(int &i, int l, int r) {
    ret[o].cnt = 0;
    i = o ++;
    if(l < r) {
        int m = (l + r) >> 1;
        build(ret[i].lc, l, m);
        build(ret[i].rc, m+1, r);
    }
}

int query(int rt, int pos, int k, int l, int r) {
    if(l == r) {
        return k + ret[rt].cnt;        //注意要加上前为值储存的数字的个数
    }
    int m = (l + r) >> 1;
    if(pos <= m) {
        //向右子树中统计，因为时统计pos之后数字的个数，所以要加上右子树所包含的数字的个数
        return query(ret[rt].lc, pos, k+ret[ret[rt].rc].cnt, l, m);
    }
    else {
        return query(ret[rt].rc, pos, k, m+1, r);
    }
}

void update(int &i, int pos, int val, int l, int r) {
    ret[o] = ret[i];
    ret[o].cnt += val;
    i = o++;
    if(l < r) {
        int m = (l + r) >> 1;
        if(pos <= m) {
            update(ret[i].lc, pos, val, l, m);
        }
        else {
            update(ret[i].rc, pos, val, m+1, r);
        }
    }
}

int main(int argc, char const *argv[]) {
    int n, m;
    while(~scanf("%d", &n)) {
        o = 0;
        build(root[0], 1, n);
        map<int, int> mp;
        int x;
        for(int i = 1; i <= n; ++ i) {
            scanf("%d", &x);
            if(mp.count(x) == 0) {
                update(root[i] = root[i-1], i, 1, 1, n);    //
            }
            else {
                // 将先前一次出现的x的信息删除
                update(root[i] = root[i-1], mp[x], -1, 1, n);
                update(root[i],  i, 1, 1, n);
            }
            mp[x] = i;    // 更新最近一次出现x的位置
        }
        scanf("%d", &x);
        while(x --) {
            int l, r;
            scanf("%d%d", &l, &r);
            // 统计以r为根节点的树中l之后的数字个数
            printf("%d
", query(root[r], l, 0, 1, n));
        }    
    }
    return 0;
}