P2664 树上游戏
对于颜色 \(c\)，如果我们把颜色 \(c\) 的点全部都删除，那么我们会得到若干个连通块。
连通块里面的路径是没有贡献的，连通块联通外面的路径都会有这个颜色做了贡献。
对于一个连通块，其里面所有点都能有 \(n-siz(连通块)\) 的贡献。
如果我们每次枚举颜色，再计算连通块，这样是 \(O(n^2)\).
那么我们考虑一次把所有颜色计算出来。
若当前 dfs 到一个 \(u\)，其颜色是 \(c\)，那么删除它，其所有儿子 \(v_1,v_2,...\) 都会各自形成一个连通块。
遍历其儿子。我们现在要求出这个儿子 \(v\) 所在连通块的大小。
如何求呢？这个连通块的大小是 \(v\) 的子树大小减去 \(v\) 子树内极大的、颜色也是 \(c\) 的子树大小。
用一个 vector 存储现在颜色是 \(c\) 的极大子树，从后往前遍历属于 \(v\) 子树里的子树。
那么