[SDOI2014]重建

一、题目

点此看题

二、解法

本体考察了矩阵树的扩展,他不仅仅能求生成树个数,其实它的本质是求所有生成树的边权乘积的和,生成树个数是边权为 1 1 1的特殊情况罢了。

形式化地说,矩阵树定理求的是这个:
∑ T ∏ e ∈ T p e \sum_{T}\prod_{e\in T}p_e TeTpe但是本题要求只选 n − 1 n-1 n1条边,每条边有 p e p_e pe可能出现,所以求的是这个:
∑ T ( ∏ e ∈ T p e ∏ e ∉ T ( 1 − p e ) ) \sum_{T}(\prod_{e\in T} p_e\prod_{e\not\in T}(1-p_e)) T(eTpeeT(1pe))做一下等价变换:
∏ ( 1 − p e ) ( ∑ T ∏ e ∈ T p e 1 − p e ) \prod (1-p_e)(\sum_{T}\prod_{e\in T}\frac{p_e}{1-p_e}) (1pe)(TeT1pepe)重新定义这个概率,然后就是矩阵树定理的板子了。

#include 
#include 
#include 
#include 
using namespace std;
const int M = 55;
#define db double
int read()
{
    int num=0,flag=1;
    char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int n;db ans,x,a[M][M];
void guass()
{
	n--;
	for(int i=1;i<=n;i++)
	{
		int mx=i;
		for(int j=i+1;j<=n;j++)
			if(a[mx][i]<a[j][i])
				mx=j;
		if(mx!=i)
		{
			for(int j=1;j<=n;j++)
				swap(a[mx][j],a[i][j]);
		}
		if(!a[i][i])
		{
			puts("0");
			return ;
		}
		for(int j=i+1;j<=n;j++)
		{
			db t=a[j][i]/a[i][i];
			for(int k=i;k<=n;k++)
				a[j][k]-=a[i][k]*t;
		}
	}
	for(int i=1;i<=n;i++)
		ans*=a[i][i];
	printf("%f\n",abs(ans));
}
signed main()
{
	n=read();ans=1;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		{
			scanf("%lf",&x);
			if(i>=j) continue;
			if(x==1) x-=1e-10;
			ans*=(1-x);
			db t=x/(1-x);
			a[i][i]+=t;a[j][j]+=t;
			a[i][j]-=t;a[j][i]-=t;
		}
	guass();
}

你可能感兴趣的:(矩阵树定理)