(转载)求生成树的个数——matrix-tree算法
题目的大意是给你一些点的坐标,然后有一个距离限制R。如果两点之间的距离小于R且他们之间没有点与他们共线就可以连通。最后要你求连通图的个数。
这个题目让我又学到了一点,那就是用矩阵树定理来计算生成树的个数。
在这里我不就证明展开讨论,因为我证明不来,感兴趣的可以看看周冬《生成树的计数及其应用》。
我就直接说定理就好了。
Matrix-Tree定理是解决生成树计数问题最有力的武器之一。
它首先于1847年被Kirchhoff证明。在介绍定理之前,我们首先明确几个概念:
1、G的度数矩阵D[G]是一个n*n的矩阵,并且满足:当i≠j时,dij=0;当i=j时,dij等于vi的度数。
2、G的邻接矩阵A[G]也是一个n*n的矩阵, 并且满足:如果vi、vj之间有边直接相连,则aij=1,否则为0。
我们定义G的Kirchhoff矩阵(也称为拉普拉斯算子)C[G]为C[G]=D[G]-A[G],
则Matrix-Tree定理可以描述为:G的所有不同的生成树的个数等于其Kirchhoff矩阵C[G]
任何一个n-1阶主子式的行列式的绝对值。所谓n-1阶主子式,就是对于r(1≤r≤n),
将C[G]的第r行、第r列同时去掉后得到的新矩阵,用Cr[G]表示
那么我们要做的就是根据题目给出的数据来建立相应的矩阵,然后求出这个矩阵的n-1阶行列式的绝对值就OK了。
另外为了优化一下算法,我们可以先DFS,看这个图是否能够连通,如果不能连通,则字节输出-1,后面就没必要在运算了。
但是事实远比我们想象的要复杂。我个人认为这题目最麻烦的就是求行列式。
我采用的是通过高斯消元法将矩阵化成上三角矩阵,然后求对角线的乘积。
我们可以看到,题目中说数据很大,所以我们要进行取模运算。
而且在对矩阵的操作中,如果交换行,则行列式的值就变成了相反数
如果乘以了某个系数,则行列式的值也相应的乘以这个系数。
我的思路来自于cyberzhg的博客。就是用一个系数来记录乘以的系数,最后想办法除掉就行了。
另外在消元的时候我们不能按照以往我们在线性代数中的那样直接除,这样会产生浮点数。所以我们采用乘以逆元的做法来消元。
当然这种做法就不可避免的乘上系数,所以我们就要记录了。
在最后除以系数的时候也有个要注意的地方,在这里再次感谢cyberzhg。下面是他给我的解答。
比如因为初等行变换,行列式的值被扩大了7倍,
最终算出来行列式的值是1,这时的1明显是因为太大取余后得到的1,
这时要除以7就会出现小数,但实际的值必然是一个整数。
假设除以7后应该得到的取余后的数是x,那么x*7取余后一定等于1,
这部做的就是枚举x,因为x也是取余后的值,不可能大于MOD。
所以在我的代码中才会在最后出现了那样的操作。至于换行的话,我们只要记住到底换了几次。如果是奇数次的话,就要取反,也就是用MOD-ans。
网上还有一些其它大神的代码,但是我觉得这个比较容易理解,适合小牛和水牛,比如说我。我就是水牛。下面上代码: