BZOJ3534 [Sdoi2014]重建 【矩阵树定理】

题目

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

输入格式

输入的第一行包含整数N。
接下来N行,每行N个实数,第i+l行,列的数G[i][j]表示城市i与j之
间仍有道路联通的概率。
输入保证G[i][j]=G[j][i],且G[i][j]=0;G[i][j]至多包含两位小数。

输出格式

输出一个任意位数的实数表示答案。
你的答案与标准答案相对误差不超过10^(-4)即视为正确。

输入样例

3

0 0.5 0.5

0.5 0 0.5

0.5 0.5 0

输出样例

0.375

提示

1 < N < =50

数据保证答案非零时,答案不小于10^-4

题解

矩阵树定理:
一个图的生成树个数等于矩阵G的(n - 1)阶行列式的值
其中矩阵G = D - A
其中D为度数矩阵,只有(i == j)的地方不为0,为(i)的度数
其中A为邻接矩阵

由基尔霍夫定理,我们直接求(n - 1)阶行列式实际上得到的是这个东西:

[sumlimits_{Tree} prodlimits_{e in Tree} w_e ]

即所有可能的生成树中各边权的积之和

也就是说,我们直接求是这个:

[sumlimits_{Tree} prodlimits_{e in Tree} P_e ]

但我们要求的答案是这个:

[sumlimits_{Tree} prodlimits_{e in Tree} P_e prodlimits_{e otin Tree} (1 - P_e) ]

我们还差后面那一串,怎么办?
考虑对所有边,令答案乘一个

[prodlimits_{e} (1 - P_e) ]

我们就得到了这个:

[sumlimits_{Tree} prodlimits_{e in Tree} P_e * (1 - P_e) prodlimits_{e otin Tree} (1 - P_e) ]

我们只需要把(prodlimits_{e in Tree} P_e * (1 - P_e))变成(prodlimits_{e in Tree} P_e)就可以了
即令

[prodlimits_{e in Tree} w_e = prodlimits_{e in Tree} frac{P_e}{1 - P_e} ]

那么我们令原矩阵A中(A_{i,j} = frac{P_{i,j}}{1 - P_{i,j}})

问题就圆满解决了

考虑除法问题:
如果一个值等于(0),就令其等于(eps)

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define eps 1e-9
#define LL long long int
#define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define BUG(s,n) for (int i = 1; i <= (n); i++) cout<<s[i]<<' '; puts("");
using namespace std;
const int maxn = 55,maxm = 100005,INF = 1000000000;
double A[maxn][maxn],G[maxn][maxn],ans = 1;
int n;
void gause(){
	for (int i = 1; i < n; i++){
		int j = i;
		for (int k = i + 1; k <= n; k++)
			if (fabs(A[k][i]) > fabs(A[j][i]))
				j = k;
		if (j != i) for (int k = i; k <= n; k++) swap(A[i][k],A[j][k]);
		if (fabs(A[i][i]) < eps){
			ans = 0;
			return;
		}
		for (j = i + 1; j <= n; j++){
			double t = -A[j][i] / A[i][i];
			for (int k = i; k <= n; k++)
				A[j][k] += A[i][k] * t;
		}
	}
}
int main(){
	scanf("%d",&n);
	REP(i,n) REP(j,n){
		scanf("%lf",&G[i][j]);
		if (fabs(G[i][j]) < eps) G[i][j] = eps;
		if (fabs(1 - G[i][j]) < eps) G[i][j] = 1 - eps;
	}
	REP(i,n) REP(j,n){
		A[i][j] = G[i][j] / (1 - G[i][j]);
		if (i < j) ans *= (1 - G[i][j]);
	}
	REP(i,n){
		A[i][i] = 0;
		REP(j,n) if (j != i) A[i][i] -= A[i][j];
	}
	gause();
	REP(i,n - 1) ans *= A[i][i];
	printf("%.10lf",fabs(ans));
	return 0;
}

原文地址:https://www.cnblogs.com/Mychael/p/8810428.html