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\n", query(root[r], l, 0, 1, n));
		}	
	}
	return 0;
}