DTOJ 4656. 「CSP-S 2019」树的重心

题意

小简单正在学习离散数学，今天的内容是图论基础，在课上他做了如下两条笔记：

• 一个大小为 $n$ 的树由 $n$ 个结点与 $n-1$ 条无向边构成，且满足任意两个结点间有且仅有一条简单路径。在树中删去一个结点及与它关联的边，树将分裂为若干个子树；而在树中删去一条边（保留关联结点，下同），树将分裂为恰好两个子树。

• 对于一个大小为 $n$ 的树与任意一个树中结点 $c$，称 $c$ 是该树的重心当且仅当在树中删去 $c$ 及与它关联的边后，分裂出的所有子树的大小均不超过 $\lfloor \frac{n}{2}\rfloor$ （其中 $\lfloor x\rfloor$ 是下取整函数）。对于包含至少一个结点的树，它的重心只可能有 $1$ 或 $2$ 个。

课后老师给出了一个大小为 $n$ 的树 $S$，树中结点从 $1\sim n$ 编号。小简单的课后作业是求出 $S$ 单独删去每条边后，分裂出的两个子树的重心编号和之和。即：

$$\sum _{(u,v)\in E}\left(\sum _{\genfrac{}{}{0ex}{}{1\le x\le n}{\text{且}x\text{号点是}{S}_u^{\prime }\text{的重心}}}x+\sum _{\genfrac{}{}{0ex}{}{1\le y\le n}{\text{且}y\text{号点是}{S}_v^{\prime }\text{的重心}}}y\right)$$

上式中，$E$ 表示树 $S$ 的边集，$(u,v)$ 表示一条连接 $u$ 号点和 $v$ 号点的边。$S_u^{\prime }$ 与 $S_v^{\prime }$ 分别表示树 $S$ 删去边 $(u,v)$ 后，$u$ 号点与 $v$ 号点所在的被分裂出的子树。

小简单觉得作业并不简单，只好向你求助，请你教教他。

测试点编号	$n=$	特殊性质
$1\sim 2$	7	无
$3\sim 5$	199	无
$6\sim 8$	1999	无
$9\sim 11$	49991	A
$12\sim 15$	262143	B
$16$	99995	无
$17\sim 18$	199995	无
$19\sim 20$	299995	无

表中特殊性质一栏，两个变量的含义为存在一个 $1\sim n$ 的排列 $p_i(1\le i\le n)$ 使得：

• A：树的形态是一条链。即 $\forall 1 \leq i \lt n $，存在一条边 $(p_i,p_i+1)$

• B：树的形态是一个完美二叉树。即 $\forall 1\le i\le \frac{n-1}{2}$ ，存在两条边 $(p_i,p_{2i})$ 与 $(p_i,p_{2i+1})$

对于所有测试点：$1\le T\le 5,1\le u_i,v_i\le n$ 。保证给出的图是一个树。

题解

考场上想的是直接枚举每个点算它作为重心的贡献，换根一下变为dfs序和子树大小的限制，难以在$O(nlogn)$内维护。

考虑不要换根，则讨论断掉的边在其子树内或子树外，在子树外可用树状数组维护子树大小的前缀和个数，而在子树内不太好做。又注意到重心的性质，直接以重心为根，割在子树内则不可能，就做完了。