小x与树 题解

小 $x$与树 题解

题目大意：

有一棵树，每个结点上有一个颜色。现在要求以$b$为根的子树中有多少种颜色。

子任务$1$：

子任务$1$的树是一条链，我们可以直接递归和桶来过这道题。

子任务$2$：

子任务$2$是$n=m$且所有结点的颜色不同。其实可以转化为：求以$b$为根的子树中有多少个结点。
这个就是一个树型动规的问题，我们假设$f_i$表示以$i$为根的子树中有多少个结点。
则$f_i={\sum }_{j\in so{n}_i}{f}_j+1$。
上面的式子表示以$i$结点为根的子树的结点个数等于它的所有儿子结点的个数的和再加$1$（它本身）。

子任务$3$：

可以暴力来做，时间复杂度为$O(nq)$。

子任务$4$：

也是用树型动规。设$f_{i,j}$表示以$i$为根的子树中有颜色$j$的结点个数。则
$f_{i,j}={\sum }_{k\in so{n}_i}{f}_{k,j}(j\ne a_i)$
$f_{i,a_i}={\sum }_{j\in so{n}_i}{f}_{j,a_i}+1$
上面的式子表示以$i$为结点的子树中有$j$颜色的个数等于它的儿子有颜色$j$的个数的和。如果$j=a_i$，那么就还要加上$1$。
那么我们就可以用$O(nm)$的时间卡过这题。