选择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