(CSP2019模拟)五彩树

题意

给定一颗$n$个节点的树，每个节点有$[1,m]$中的一种颜色，求一个包含不少于$k$种颜色的连通块，使其它节点到该连通块最短路径的最大值最大。
数据范围：$n,m\le 1000000$

题解

如果已经确定一个到连通块最短路径最大的节点$p$，那么答案就是以$p$为根，最深的包含至少$k$中颜色的点的子树。求子树的颜色种数可以用将同种颜色按一定顺序，两两之间的$lca$打上$-1$的标记，本身打上$+1$的标记，计算子树的和来维护（考场上没想到这个就凉了）。于是考虑枚举根，设$f[u]$为$u$子树内包含至少$k$种颜色的最深子树，进行换根DP。在换根时主要问题在于旧的根变为新的子树时，其本身的贡献要重新计算，即它是否包含至少$k$种颜色，而这个新的子树不是原来树中存在的，考虑转化为它的补集（即要换根的子树），而条件变为不完全包含的颜色不少于$k$种，对应到$dfs$序上则是一个二维偏序问题，用树状数组维护即可。