「SDOI2014」重建 解题报告

「SDOI2014」重建

题意

给一个图\(G\),两点\((u,v)\)有边的概率是\(p_{u,v}\),求有\(n-1\)条边通行且组成了一颗树的概率是多少。


抄了几个矩阵树定理有趣的感性说法

  • 矩阵树定理的度数矩阵记录的是每个点的边权和,邻接矩阵记录的是边权,求的则是所有生成树的边权乘积和

  • 考虑Kirchhoff矩阵的意义:\(K[G]=D[G]−A[G]=B[G]B^T[G]\),之所以能够进行生成树计数是对于其伴随矩阵在计数\(n−1\)条边的集合时,当\(n−1\)条边中存在环就会产生线性组合而导致行列式为零,否则恰好对角线上均为伴随矩阵中所赋的值,使得\(\det(B_{i,j})^2\)就为\(1\)

考虑直接把度数矩阵赋为出度概率和,连边矩阵为概率,然后相减套矩阵树定理求得是什么

\[\sum_T\prod_{(u,v)\in T}p_{u,v} \]

然而我们需要求

\[\sum_T\prod_{(u,v)\in T}p_{u,v}\prod_{(u,v)\notin T}(1-p_{u,v}) \]

化一下可以得到

\[\prod_{(u,v)\in G}(1-p_{u,v})\sum_T\prod_{(u,v)\in T}\frac{p_{u,v}}{1-p_{u,v}} \]

然后把后面的拿去跑矩阵树就可以了。

注意一些精度问题,把\(p=0\)搞成\(p=\epsilon\)\(p=1\)搞成\(1-\epsilon\)差不多就可以了


Code:

#include <cstdio>
#include <cmath>
#include <algorithm>
const int N=52;
const double eps=1e-10;
int n;
double p[N][N],a[N][N];
void Gauss()
{
    --n;
	for(int i=1;i<=n;i++)
	{
		int id=i;
		for(int j=i+1;j<=n;j++)
			if(fabs(a[id][i])<fabs(a[j][i])) id=j;
		std::swap(a[id],a[i]);
		for(int j=i+1;j<=n;j++)
		{
		    double p=a[j][i]/a[i][i];
		    for(int k=n;k>=i;k--)
				a[j][k]-=a[i][k]*p;
		}
	}
}
int main()
{
	scanf("%d",&n);
	double sum=1;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		{
			scanf("%lf",&p[i][j]);
			if(p[i][j]==0) p[i][j]=eps;
			if(p[i][j]==1) p[i][j]=1-eps;
			if(i<j) sum*=1-p[i][j];
			a[i][j]=p[i][j]/(1-p[i][j]);
		}
	for(int i=1;i<=n;i++)
	{
	    a[i][i]=0;
	    for(int j=1;j<=n;j++)
			if(i!=j)
				a[i][i]+=a[i][j],a[i][j]=-a[i][j];
	}
	Gauss();
	for(int i=1;i<=n;i++) sum*=fabs(a[i][i]);
	printf("%.4lf\n",sum);
	return 0;
}

2019.2.21

posted @ 2019-02-21 20:36  露迭月  阅读(303)  评论(0编辑  收藏  举报