算法记录 004： 矩阵树(Matrix Tree)和其扩展

实用算法 004： 矩阵树(Matrix Tree)和其扩展

推荐知乎好文

矩阵树可以计算一个无向图的生成树个数或有向图的根向生成树和叶向生成树的个数。

由于有向图不怎么用到，下面只讨论无向图的情况。

注意：无向图可以存在重边，但不能存在自环

在下面的内容中默认N是点数，M是边数

定义1. 邻接矩阵(M)，其中(M_{i,j})表示(i,j)之间的边数。可以发现(M)是一个对称且对角线是(0)的(N)阶矩阵。

定义2. 度数矩阵(D)，其中(D_{i,i})表示(i)的度数，若(i eq j)则(D_{i,j}=0)。

定义3. 拉普拉斯矩阵（Laplacian matrix）(L=D-M)

其中一个无向图的生成树个数(=det L_{0}) （其中(L_0)表示去掉(L)的第(i)列和第(i)行得到的矩阵，(i)任意），证明知乎的文章里有。

则对于一个图，算出生成树个数的时间复杂度为(O(N^3))，主要是高斯消元算行列式的复杂度。

扩展

如果无向图的边有边权，然后可以令(M_{i,j})表示(i,j)之间边的边权(没有边则为0)。

然后(det L_0)就是=

[sum_{T} prod_{{e_i}in T} w_i ]

也就是所有生成树边的乘积的和。

(proof:)

将权值理解成边数，也就是(i,j)之间有了(M_{i,j})条边。

特殊的： 边权不要求是数，甚至可以是多项式（如例题）

例题:

CF 917 D

这题是要求算出与原树有恰好(k)条边重合的方案数。

我们可以构建一张图，图中边权定义如下，边权为多项式。

• 若边(i,j)是原树中的边，则边权为(x^1)。

• 否则为(x^0)

然后通过矩阵树定理，算出的(det L_0)也是一个多项式。

则和原树有恰好k条边重合的方案数即为(x^k)的系数。

但是这样直接套ntt,时间复杂度为(O(n^4log n)),优化方法见：003