无向图生成树计数 -- Kirchhoff 矩阵法模板

Kirchhoff 矩阵法是根据Matrix-Tree定理来的，本人太菜，没有那个心力去看证明了，知道是用就可以了

作用：

给定一个n个点m条边的无向图，求出这个图的生成树的总数。

Matrix-Tree定理(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]表示。

矩阵树方法实现：
实现方法很简单，第一步是构建拉氏矩阵，很简单。难点在于实现求行列式的值。我这里采用矩阵初等变换将矩阵转化为上三角矩阵，这样行列式的值就等于主对角元素乘积。我实现了打印拉氏矩阵C和输出图生成树个数这两个方法，主体程序如下：

#include<bits/stdc++.h>

using namespace std;

class spanningTreeNum {
private:
    int V = 0;  // 顶点数
    vector<vector<int> > c; // 拉式矩阵c=d-A
public:
    spanningTreeNum(int V)
    {
        this->V = V;
        c = vector<vector<int> >(V, vector<int>(V, 0));  //初始化二维矩阵c为0
    }
    void addEdge(int u, int v);  // u和v之间加一条边
    int getTreeNum();
    int det(vector<vector<float> > A);  // 求行列式A的值
    void showC();
};

void spanningTreeNum::addEdge(int u, int v) {
    c[u][u]++;  // 顶点度数加一
    c[v][v]++;
    c[u][v] = -1;  // 表示顶点u、v之间有一条边，因为c=d-A，所以为-1
    c[v][u] = -1;
}

int spanningTreeNum::det(vector<vector<float> > A) {
    /*
     * 思路是将利用初等变换A转化为上三角矩阵，这样对角线元素乘积即为行列式值
     */
    float res = 1;
    int iter = 0;  // 记录交换次数
    for (int i = 0; i < A.size(); ++i) {  //该for循环内的逻辑是将矩阵转化为上三角矩阵
        if(A[i][i]==0) 
        {
            for (int j = i; j < A.size(); ++j) {
                if(A[j][i]!=0) {
                    swap(A[i], A[j]);
                    iter++;
                }
            }
        }
        for (int j = i+1; j < A.size(); ++j) {
            float temp = -A[j][i]/A[i][i];
            for (int k = 0; k < A[j].size(); ++k) {
                A[j][k] = A[i][k]*temp+A[j][k];
            }
        }
    }
    for (int i = 0; i < A.size(); ++i) {
        res *= A[i][i];
    }
    if(iter%2==1) res = -res;
    return (int)res;
}
int spanningTreeNum::getTreeNum() {
    // 求余子式
    vector<vector<float > > temp(V-1, vector<float >(V-1, 0));
    for (int i = 1; i < V; ++i) {
        for (int j = 1; j < V; ++j) {
            temp[i-1][j-1] = c[i][j];
        }
    }
    return det(temp);
}

void spanningTreeNum::showC() {
    for (int i = 0; i < c.size(); ++i) {
        for (int j = 0; j < c[i].size(); ++j) {
            printf("%3d ", c[i][j]);
        }
        cout << endl;
    }
}

int main()
{
    spanningTreeNum G(4);
    G.addEdge(0, 1);
    G.addEdge(0, 2);
    G.addEdge(1, 2);
    G.addEdge(2, 3);

    cout << "拉氏矩阵为：" << endl;
    G.showC();  //打印拉式矩阵C
    cout << "生成树个数为：" << G.getTreeNum() << endl; // 打印生成树个数
    return 0;
}

运行结果：