题解：P10842 【MX-J2-T3】Piggy and Trees

题目链接:

• link，点击这里喵。

题意简述:

• 给出一棵树，取一点 $i$，并另取两点 $u$ 和 $v$，在 $u$ 和 $v$ 简单路径上任取一点使这点与 $i$ 的距离最小，计算出这样所有的组合的距离的总和。

思路:

• 看眼数据范围，总点数 $n\le 2\times 10^5$，考虑线性算法，有关于树，首先考虑树上 dp。
• 这道题题面很具有迷惑性，她把 $\sum _{u=1}^n\sum _{v=u+1}^n\sum _{i=1}^{n}f(u,v,i)$ 中的 $\sum\limits_{i = 1}^n $ 放在了后面，这就导致如果思路想到枚举 $u$ 和 $v$，就很难 dp 了，根据 $\sum$ 的性质，我们把 $\sum\limits_{i = 1}^n $ 提到外面去，原式就变成了 $\sum _{i=1}^n\sum _{u=1}^n\sum _{v=u+1}^{n}f(u,v,i)$，思路就简洁明了了，记录每个点的答案，最后相加即可。
• 那么 $O(n)$ 做法考虑换根 dp，这里假设以点 $1$ 为出发点。

做法:

• 定义 $l{j}_i$ 为点 $i$ 及其子树所包含的点对数量，$si{z}_i$ 为点 $i$ 及其子树的大小，$si{z}_i$ 为以点 $i$ 为 $\sum _{u=1}^n\sum _{v=u+1}^{n}f(u,v,i)$ 的答案。

  dfs1:

  • $si{z}_{now}=$