【LOJ2983】「WC2019」数树

最新推荐文章于 2020-03-08 16:51:28 发布

cz_xuyixuan

最新推荐文章于 2020-03-08 16:51:28 发布

阅读量681

点赞数 1

分类专栏：【类型】做题记录【OJ】LOJ 【算法】组合数学【算法】动态规划【算法】生成函数【算法】树的Prufer序列【算法】容斥原理【算法】微积分【算法】FFT与NTT 【算法】CDQ分治与整体二分

本文链接：https://blog.csdn.net/qq_39972971/article/details/88935680

版权

这篇博客详细解析了LOJ2983数树问题的解决方案，包括三种操作（op=0, 1, 2）的思路和优化算法。针对op=1，通过Prufer序列和树形DP实现O(N)复杂度；对于op=2，利用分治和NTT优化达到O(NLog2N)。同时，介绍了如何运用多项式exp计算的方法。" 110450326,9552608,Eigen安装错误与Ceres冲突解决,"['Eigen库', 'ceres库', 'Linux环境', '错误处理', '软件安装']

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【题目链接】

点击打开链接

【思路要点】

$o p = 0$ ，算一算两棵树的公共边数即可。

时间复杂度 $O (N)$ 或 $O (N L o g N)$ 。

$o p = 1$ ，考虑枚举蓝树上的一个边集 $S$ ，强制红树上同样存在这些边，计算将剩余 $N - ∣ S ∣$ 个联通块连成一棵树的方案数，更新答案。

上述算法中，枚举蓝树上的一个大小为 $S$ 的边集后计算将剩余 $N - S$ 个联通块连成一棵树的方案数时，一个包含恰好 $T$ 条蓝边的红树会被计算 $\binom{T}{S}$ 次，并且由于最终的生成树上的蓝边数尚不确定，我们也无法得知需要乘上 $y$ 的多少次方。

注意到 $y^{-T}=(y^{-1}-1+1)^T=\sum_{S=0}^{T}(y^{-1}-1)^S\binom{T}{S}$ ，若看做每选取一条蓝边产生 $\times(y^{-1}-1)$ 的贡献，我们就可以直接应用上述算法，只需将最终答案乘以 $y^N$ 即可。

由 $P r u f e r$ 序列，将 $M$ 个大小分别为 $a_1,a_2,a_3,...,a_M\ (\sum a_i=N)$ 的联通块连成一棵树的方案数为 $N^{M-2}\prod a_i$ 。 $N^{M-2}$ 部分的贡献可以看做每选取一条蓝边产生 $×N−1 \times N^{-1}$ 的贡献，最后将答案乘以 $N^{N-2}$ 。 $\prod a_i$ 部分的贡献对应的组合意义为在每一个联通块内选择恰好一个代表点的方案数。

至此，我们可以设计一个简单树形 $d p$ 来加速上述枚举算法。

即记 $dp_{i,0/1}$ 表示 $i$ 所在联通块有/没有选出代表点的情况下，其子树内上述算法所有情况的贡献总和，可以 $O (1)$ 简单转移。

时间复杂度 $O (N)$ 。

$o p = 2$ ，由

最低0.47元/天解锁文章