CF2152G

有一棵以 \(1\) 为根, \(n\) 个节点的树,每个节点有一个颜色白/黑。给定 \(q\) 组询问,每组询问给了一个 \(u\),表示将 \(u\) 子树内的点的颜色全部翻转。每次操作后回答至少需要几条从根开始的链才能覆盖所有黑点覆盖。

先来转化一下问题,题目问的其实就是有多少个节点 \(u\) 满足 \(u\) 为黑点且 \(u\) 的子树内没有其他黑点。

\(u\) 的子树在 dfs 序上对应 \([dfn_u, bk_u]\)\(nxt_u\) 表示 \(u\)dfs 序上后面第一个黑点的位置,\(co_u\) 表示 \(u\) 的颜色。答案就是满足 \(co_u = 1, nxt_u > bk_u\)\(u\) 的数量。

这就可以用线段树维护了,线段树维护 dfs 序上 \([l, r]\) 这些点里最靠左的黑点,最靠右的黑点以及满足要求的 \(u\) 的数量。合并时只有左半区间最靠右的黑点可能从满足变成不满足,减一下即可。

struct SegTree {
	int l, r, c; // 最靠左、靠右的黑点,满足条件的黑点数。
} tr[2][MAXN * 4];

void Pushup(SegTree *tr, int u, int l, int r) {
	tr[u] = {(tr[l].l ? tr[l].l : tr[r].l), (tr[r].r ? tr[r].r : tr[l].r), tr[l].c + tr[r].c};
	tr[u].c -= (tr[l].r && tr[r].l && tr[r].l <= bk[vis[tr[l].r]]); // 减去 tr[l].r 不合法的情况。
}

再考虑翻转,我们对白点也做同样的事情,碰到被 \([dfn_u, bk_u]\) 包含的区间 swap 一下白点和黑点的信息即可。

时间复杂度:\(O(n + q\log n)\)

主要是把题目条件转化成可以用线段树维护形式就可以了。

posted @ 2025-10-24 12:09  xiehanrui0817  阅读(15)  评论(0)    收藏  举报