bzoj 3534: [Sdoi2014]重建 矩阵树定理

题意

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

分析

显然这题可以用矩阵树定理来统计，关键在于如何建矩阵。
一般的是度数矩阵-邻接矩阵，求出来的东西可以看成是生成树上每条边的权值乘起来。在一般的生成树计数里边权可以看作是1。
这题我们要求$\prod_{(x,y)\in E}\frac{p_{(x,y)}}{1-p_{(x,y)}}*\prod_{(x,y)\not\in E}(1-p_{(x,y)})$
那么我们可以把$\frac{p_{(x,y)}}{1-p_{(x,y)}}$作为边权，答案就是行列式*上面式子的右边。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;

const int N=55;
const double eps=1e-8;

int n;
double a[N][N];

double gauss(int n)
{
    double ans=1;
    for (int i=1;i<=n;i++)
    {
        int l=i;
        for (int j=i+1;j<=n;j++)
            if (fabs(a[j][i])>fabs(a[l][i])) l=j;
        if (l!=i) for (int j=i;j<=n;j++) swap(a[i][j],a[l][j]);
        if (fabs(a[i][i])<eps) return 0;
        for (int j=i+1;j<=n;j++)
        {
            if (fabs(a[j][i])<eps) continue;
            double w=a[j][i]/a[i][i];
            for (int k=i;k<=n;k++) a[j][k]-=a[i][k]*w;
        }
        ans*=a[i][i];
    }
    return fabs(ans);
}

int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
            scanf("%lf",&a[i][j]);
    double tmp=1;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
            if (i!=j&&fabs(a[i][j])>eps)
            {
                double w=fabs(1-a[i][j])<eps?eps:1-a[i][j];
                if (i<j) tmp*=w;
                a[i][j]=-a[i][j]/w;
                a[i][i]-=a[i][j];
            }
    printf("%.8lf",gauss(n-1)*tmp);
    return 0;
}