[PKUWC2019]Day1 T2 你和虚树的故事

选择k个颜色，使得颜色的虚树有交的方案数

肯定要考虑连通块的贡献。

法一

https://www.cnblogs.com/xzz_233/p/10292983.html

枚举连通块还是不可行的。

枚举连通块的最上面的点

建立虚树，差分，得到每个点作为虚树最高点的虚树个数ai，在另外bi个虚树的内部

这个点i作为最高点贡献的连通块个数可以计算

和ai,bi密切相关，开个桶存储个数，枚举ai+bi（或者说bi）的个数，NTT

法二

http://blog.leanote.com/post/rockdu/0347

更巧妙的方法

一个连通块的贡献为1，1=交集点数-交集边数

所以统计所有k个颜色虚树交出的点的个数-交出的边的个数

然后枚举每个点的贡献，开个桶，还是NTT