SDOI 2014 重建 题解

题目传送门

题目大意： 给出一张图，地震过后每条边有一个存在的概率，问地震后图变树的概率。

题解

又长见识了……还有个变元矩阵树定理。

对于图的一棵生成树，它的出现概率为：存在的边的 $p[i]$ 之积乘不存在的边的 $(1-p[i])$ 之积。设图为 $G$，生成树为 $T$，那么用柿子表示出来就是：$\prod _{e\in T}p[e]\times \prod _{e^{\prime }\notin T}(1-p[e^{\prime }])$，转换一下得到 $\prod _{e\in T}\frac{p[e]}{1-p[e]}\times \prod _{e^{\prime }\in G}(1-p[e^{\prime }])$。

转化后，一棵生成树的出现概率就变成了树中所有边的 $\frac{p[e]}{1-p[e]}$ 之积在乘上一个定值 $\prod _{e^{\prime }\in G}(1-p[e^{\prime }])$ 了，然后所有生成树出现概率总和就是答案。于是我们就可以将每条边的边权定为 $\frac{p[e]}{1-p[e]}$，然后跑一次变元矩阵树定理，最后乘上那个定值就是答案了。

代码如下：

#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
#define maxn 60
#define eps 0.000001

int n;
double f[maxn][maxn],prod=1.0;
double det(int l)
{
	double re=1;
	for(int i=1;i<=l;i++)
	{
		int p=i;
		for(int j=i+1;j<=l;j++)
		if(fabs(f[j][i])>fabs(f[p][i]))p=j;
		if(p!=i)swap(f[i],f[p]);
		
		for(int j=i+1;j<=l;j++)
		for(int k=l;k>=i;k--)
		f[j][k]-=f[j][i]/f[i][i]*f[i][k];
		re*=f[i][i];
	}
	return re;
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	for(int j=1;j<=n;j++)
	scanf("%lf",&f[i][j]);
	for(int i=1;i<=n;i++)
	for(int j=1;j<=n;j++)
	if(i!=j)
	{
		double p=1.0-f[i][j]<eps?eps:1.0-f[i][j];
		if(i<j)prod*=p; f[i][j]/=p;
		f[i][i]+=f[i][j],f[i][j]=-f[i][j];
	}
	printf("%.5lf",det(n-1)*prod);
}