bzoj4771 七彩树 dfs序+主席树+树链的并

题目传送门

https://lydsy.com/JudgeOnline/problem.php?id=4771

题解

一道不错的树链并的基础练习题。

如果不是树,而是一个数组的话,对于给定区间内的不同颜色数,我们可以维护一个 \(pre_{i, j}\) 表示前 \(i\) 个最后一个 \(j\) 出现的位置。那么答案就是 \(pre_{r, j} \geq l\)\(j\) 的个数,用主席树维护,第一维 \(i\),第二维 \(pre_i, j\),维护的是数量就可以了。(当然这个问题还有别的方法,比如维护 \(nxt_i\) 之类的)

所以类似的,对于一个点 \(x\),维护 \(pre_{x, i}\) 表示从 \(1\)\(x\) 的链上,满足条件的颜色必须 \(pre_{y, i} \in subtree_x\),最后 \(i\) 颜色最后一次出现的位置,同样第一维 \(i\),第二维 \(pre_i, j\),维护的是数量。

但是这个做法显然有问题。如果 \(x\) 的子树中第 \(d\) 层的点数不止一个,那么会有很多重复的。

发现每一个颜色的点都会对从 \(1\) 到自己的链上的点产生贡献,但是一种颜色只能产生一种贡献。所以考虑进行树链的并来维护就可以了。这样做的第一维就是层数而不是编号了。


时间复杂度 \(O(n\log n)\)

这道题不卡常但是因为我滥用 memset 导致当做卡常题调了半天。

线段树的点数极限很大,但是跑不满,所以不要直接把整个线段树 memset 一遍。


#include<bits/stdc++.h>

#define fec(i, x, y) (int i = head[x], y = g[i].to; i; i = g[i].ne, y = g[i].to)
#define dbg(...) fprintf(stderr, __VA_ARGS__)
#define File(x) freopen(#x".in", "r", stdin), freopen(#x".out", "w", stdout)
#define fi first
#define se second
#define pb push_back

template<typename A, typename B> inline char smax(A &a, const B &b) {return a < b ? a = b , 1 : 0;}
template<typename A, typename B> inline char smin(A &a, const B &b) {return b < a ? a = b , 1 : 0;}

typedef long long ll; typedef unsigned long long ull; typedef std::pair<int, int> pii;

template<typename I> inline void read(I &x) {
	int f = 0, c;
	while (!isdigit(c = getchar())) c == '-' ? f = 1 : 0;
	x = c & 15;
	while (isdigit(c = getchar())) x = (x << 1) + (x << 3) + (c & 15);
	f ? x = -x : 0;
}

const int N = 1e5 + 7;

int n, m, nod, dfc, cnt;
int c[N], rt[N], q[N];
int dep[N], f[N], siz[N], son[N], dfn[N], pre[N], top[N];

struct Node { int lc, rc, val; } t[N * 70];
inline void ins(int &o, int L, int R, int x, int k) {
	++cnt;
	t[++nod] = t[o], t[o = nod].val += k;
	if (L == R) return;
	int M = (L + R) >> 1;
	if (x <= M) ins(t[o].lc, L, M, x, k);
	else ins(t[o].rc, M + 1, R, x, k);
}
inline int qsum(int o, int L, int R, int l, int r) {
	++cnt;
	if (!o) return 0;
	if (l <= L && R <= r) return t[o].val;
	int M = (L + R) >> 1;
	if (r <= M) return qsum(t[o].lc, L, M, l, r);
	if (l > M) return qsum(t[o].rc, M + 1, R, l, r);
	return qsum(t[o].lc, L, M, l, r) + qsum(t[o].rc, M + 1, R, l, r);
}

inline void qadd(int &o, int x, int k) { /*dbg("x = %d, k = %d\n", x, k);*/ ins(o, 1, n, dfn[x], k); }
inline int qval(int o, int x) { return qsum(o, 1, n, dfn[x], dfn[x] + siz[x] - 1); }

struct Edge { int to, ne; } g[N << 1]; int head[N], tot;
inline void addedge(int x, int y) { g[++tot].to = y, g[tot].ne = head[x], head[x] = tot; }
inline void adde(int x, int y) { addedge(x, y), addedge(y, x); }

inline void dfs1(int x, int fa = 0) {
	dep[x] = dep[fa] + 1, f[x] = fa, siz[x] = 1;
	for fec(i, x, y) if (y != fa) dfs1(y, x), siz[x] += siz[y], siz[y] > siz[son[x]] && (son[x] = y);
}
inline void dfs2(int x, int pa) {
	top[x] = pa, dfn[x] = ++dfc, pre[dfc] = x;
	if (!son[x]) return; dfs2(son[x], pa);
	for fec(i, x, y) if (y != f[x] && y != son[x]) dfs2(y, y);
}
inline int lca(int x, int y) {
	++cnt;
	while (top[x] != top[y]) dep[top[x]] > dep[top[y]] ? x = f[top[x]] : y = f[top[y]];
	return dep[x] < dep[y] ? x : y;
}

struct cmp {
	inline bool operator () (const int &x, const int &y) {++cnt; return dfn[x] < dfn[y]; }
};
std::set<int, cmp> s[N];

inline void ins(std::set<int, cmp> &s, int dep, int x) {
	std::set<int, cmp>::iterator p = s.lower_bound(x);
	int y = 0, z = 0;
	qadd(rt[dep], x, 1);
	if (p != s.end()) y = *p, qadd(rt[dep], lca(x, y), -1);
	if (p != s.begin()) z = *--p, qadd(rt[dep], lca(x, z), -1);
	if (y && z) qadd(rt[dep], lca(y, z), 1);
	// dbg("ins: dep = %d, x = %d, y = %d, z = %d, qry = %d, rt[dep] = %d\n", dep, x, y, z, qval(rt[dep], x), rt[dep]);
	s.insert(x);
}

inline void bfs() {
	int hd = 0, tl = 1;
	q[tl] = 1;
	while (hd < tl) {
		int x = q[++hd];
		for fec(i, x, y) if (y != f[x]) q[++tl] = y;
	}
}
inline void build() {
	for (int i = 1; i <= n; ++i) {
		int x = q[i];
		if (dep[x] != dep[q[i - 1]]) rt[dep[x]] = rt[dep[x] - 1];
		ins(s[c[x]], dep[x], x);
	}
}

inline void work() {
	dfs1(1), dfs2(1, 1);
	bfs();
	build();
	int la = 0;
	while (m--) {
		int x, d;
		read(x), read(d);
		x ^= la, d ^= la;
		// dbg("x = %d, d = %d\n", x, d);
		d += dep[x], smin(d, dep[q[n]]);
		printf("%d\n", la = qval(rt[d], x));
		// la = qval(rt[d], x);
	}
	memset(t, 0, sizeof(Node) * (nod + 1));
}

inline void cls() {
	tot = nod = dfc = 0;
	memset(head, 0, sizeof(head));
	memset(son, 0, sizeof(son));
	for (int i = 1; i <= n; ++i) s[i].clear();
}

inline void init() {
	read(n), read(m);
	cls();
	for (int i = 1; i <= n; ++i) read(c[i]);
	for (int i = 2, x; i <= n; ++i) read(x), addedge(x, i);
}

int main() {
#ifdef hzhkk
	freopen("hkk.in", "r", stdin);
#endif
	int T;
	read(T);
	while (T--) {
		init();
		work();
	}
	fclose(stdin), fclose(stdout);
	return 0;
}
posted @ 2019-10-09 11:19  hankeke  阅读(128)  评论(0编辑  收藏  举报