【BZOJ3534】重建（矩阵树定理）

【BZOJ3534】重建（矩阵树定理）

题面

BZOJ

洛谷

题解

这。。。。

矩阵树定理神仙用法？？？？

#include<iostream>
#include<cmath>
using namespace std;
#define ll long long
#define eps 1e-8
int n;
double g[55][55],s=1;
double Gauss()
{
	double ret=1;
	for(int i=1;i<n;++i)
	{
		int nw=i;
		for(int j=i+1;j<n;++j)if(fabs(g[j][i])>fabs(g[nw][i]))nw=j;
		swap(g[nw],g[i]);
		for(int j=i+1;j<n;++j)
		{
			double t=g[j][i]/g[i][i];
			for(int k=i;k<n;++k)g[j][k]-=t*g[i][k];
		}
		ret*=g[i][i];
	}
	return fabs(ret);
}
int main()
{
	ios::sync_with_stdio(false);
	cin>>n;
	for(int i=1;i<=n;++i)
		for(int j=1;j<=n;++j)
		{
			cin>>g[i][j];
			if(g[i][j]<eps)g[i][j]=eps;
			if(g[i][j]>1-eps)g[i][j]=1-eps;
			if(i<j)s*=1-g[i][j];
			g[i][j]/=1-g[i][j];
		}
	for(int i=1;i<=n;++i)
		for(int j=1;j<=n;++j)
			if(i!=j)g[i][i]-=g[i][j];
	printf("%.8lf\n",Gauss()*s);
}