LOJLOJLOJ 传送门
题解:首先考虑一棵树怎么做,就是树形 dpdpdp,然后我们可以枚举每一条边怎么选,显然有 3 种情况
进一步发现只需要枚举两种,即强制 uuu 选 vvv 不选,和强制 uuu 不选 vvv 随意
那么现在的问题就是每次 banbanban 掉一些点选一些点不选,动态更新根节点的 dpdpdp 值
然后把转移写成矩阵的形式链分治,由于可以除 000 所以手写了一个用 x∗0yx*0^yx∗0y 表示每个数的类
复杂度 O(n+k∗2k∗log(n)2)O(n+k*2