令 dpidp_idpi 表示强制 iii 的父边选后它子树内的方案数。
有一个显然的 O(∑siz)\mathcal{O}(\sum siz)O(∑siz) 的做法就是把以每个节点的子树中再 dp 一下“上次”选的点的集合得到的总方案数(根据限制,其中有某些子树不能选)。
赛后我瞎卡了卡复杂度没变但是过了。。。
求如何优化复杂度。/kel