BZOJ3534:[SDOI2014]重建(矩阵树定理)

Description

T国有N个城市，用若干双向道路连接。一对城市之间至多存在一条道路。
在一次洪水之后，一些道路受损无法通行。虽然已经有人开始调查道路的损毁情况，但直到现在几乎没有消息传回。
幸运的是，此前T国政府调查过每条道路的强度，现在他们希望只利用这些信息估计灾情。具体地，给定每条道路在洪水后仍能通行的概率，请计算仍能通行的道路恰有N-1条，且能联通所有城市的概率。

Input

输入的第一行包含整数N。
接下来N行，每行N个实数，第i+l行，列的数G[i][j]表示城市i与j之
间仍有道路联通的概率。
输入保证G[i][j]=G[j][i]，且G[i][j]=0；G[i][j]至多包含两位小数。

Output

输出一个任意位数的实数表示答案。
你的答案与标准答案相对误差不超过10^(-4)即视为正确。

Sample Input

3
0 0.5 0.5
0.5 0 0.5
0.5 0.5 0

Sample Output

0.375

HINT

1 < N < =50

数据保证答案非零时，答案不小于10^-4

Solution

题目即让求：（$T$是生成树，$e$是边）
$\sum_{T}\prod_{e\in T} p_e \prod_{e\notin T}(1-p_e)$
把第二个$\prod$变一下
$\sum_{T}\prod_{e\in T} p_e \frac{\prod_{e}(1-p_e)}{\prod_{e\in T}(1-p_e)}$
也就是
$\prod_{e}(1-p_e)\sum_{T}\prod_{e\in T} \frac{p_e}{1-p_e}$。
其中行和矩阵$-$边权矩阵的行列式的值$=\sum_{T}\prod_{e\in T} w_e$，其中$w$是边权。
然后就可以高斯消元求解了。

Code

 #include<iostream>

 #include<cstring>

 #include<cstdio>

 #include<cmath>

 #define N (59)

 #define eps (1e-10)

 using namespace std;

 int n;

 double a[N][N],f[N][N],ans=;

 void Gauss()

 {

     int w=;

     for (int i=; i<=n-; ++i)

     {

         int num=i;

         for (int j=i+; j<=n-; ++j)

             if (fabs(f[j][i])>fabs(f[num][i])) num=j;

         if (num!=i) swap(f[num],f[i]), w=-w;

         for (int j=i+; j<=n-; ++j)

         {

             double t=f[j][i]/f[i][i];

             for (int k=i; k<=n-; ++k)

                 f[j][k]-=t*f[i][k];

         }

     }

     for (int i=; i<=n-; ++i) ans*=f[i][i];

     for (int i=; i<=n; ++i)

         for (int j=i+; j<=n; ++j)

             ans*=-a[i][j];

     printf("%.10lf\n",ans*w);

 }

 int main()

 {

     scanf("%d",&n);

     for (int i=; i<=n; ++i)

         for (int j=; j<=n; ++j)

         {

             scanf("%lf",&a[i][j]);

             if (a[i][j]==) a[i][j]-=eps;

             if (i==j) continue;

             f[i][j]=-a[i][j]/(-a[i][j]);

             f[i][i]-=f[i][j];

         }

     Gauss();

 }

巴特西