生成树计数

给定无向图邻接矩阵,求该无向图生成树的个数

神奇的Matrix-Tree定理:

 

设矩阵G[N][N]为图的度数矩阵, G[i][i] = i点度数,其余元素为0

设矩阵A[N][N]为图的邻接矩阵

则定义Kirchhoff矩阵为G[N][N] - A[N][N]

 

那么该无向图的生成树个数为Kirchhoff矩阵任意n-1阶主子式的行列式值。

 

求行列式用Gauss消元就行,注意主元的选取。 时间复杂度O(n^3)

 

以下为代码,题目是CodeCraft 2007 的第四题。

参考文献可参见2007年国家集训队,周冬的论文。

  1. #include 
  2. #include 
  3. const int N = 20;
  4. double kir[N][N];
  5. int n;
  6. double det()
  7. {
  8.     int i, j, k;
  9.     double t;
  10.     double ans = 1.;
  11.     
  12.     for(i = 1; i < n; i++)
  13.     {
  14.         int tmp;
  15.         double num = 0;
  16.         for(j = i; j < n; j++)
  17.         {
  18.             if(fabs(kir[j][i]) > num)
  19.             {
  20.                 num = fabs(kir[j][i]);
  21.                 tmp = j;
  22.             }
  23.         }
  24.         if(num < 1e-9) return 0;
  25.         ans *= num;
  26.         if(tmp != i)
  27.         {
  28.             for(k = i; k < n; k++)
  29.             {
  30.                 t = kir[i][k]; kir[i][k] = kir[tmp][k]; kir[tmp][k] = t;
  31.             }
  32.         }
  33.         for(j = i + 1; j < n; j++)
  34.         {
  35.             double factor = -1.0 * kir[j][i] / kir[i][i];
  36.             for(k = i; k < n; k++)
  37.                 kir[j][k] += factor * kir[i][k];
  38.         }
  39.     }
  40.     
  41.     return ans;
  42. }
  43.                 
  44.         
  45. int main()
  46. {
  47.     int i, j, k, ca;
  48.     
  49.     for(scanf("%d",&ca); ca; ca--)
  50.     {
  51.         memset(kir, 0, sizeof(kir));
  52.         
  53.         scanf("%d",&n);
  54.         for(i = 1; i <= n; i++)
  55.             for(j = 1; j <= n; j++){
  56.                 scanf("%d",&k);
  57.                 kir[i][i] += k;
  58.                 if(i != j) kir[i][j] = -k;
  59.             }
  60.         printf("%.0lf/n", det());
  61.     }
  62.     return 0;
  63. }
  64.         
  65.     

你可能感兴趣的:(生成树计数)