CHelga

根据题意，我们可以发现这是一道树形 dp。首先考虑设计状态，注意到 $k$ 较小，那么我们可以在 dp 数组里面塞一维来维护特殊颜色点的个数。然后题目里有颜色编号的大小限制，所以第三维用 $0/1/2$ 来分别表示当前颜色小于/等于/大于 $k$ 的情况。那么这样的话就是用 $f_{i,j,0/1/ ......