题意
T国有N个城市,用若干双向道路连接。一对城市之间至多存在一条道路。
在一次洪水之后,一些道路受损无法通行。虽然已经有人开始调查道路的损毁情况,但直到现在几乎没有消息传回。
辛运的是,此前T国政府调查过每条道路的强度,现在他们希望只利用这些信息估计灾情。具体地,给定每条道路在洪水后仍能通行的概率,请计算仍能通行的道路恰有N-1条,且能联通所有城市的概率。
1 < N < =50
分析
显然这题可以用矩阵树定理来统计,关键在于如何建矩阵。
一般的是度数矩阵-邻接矩阵,求出来的东西可以看成是生成树上每条边的权值乘起来。在一般的生成树计数里边权可以看作是1。
这题我们要求
∏(x,y)∈Ep(x,y)1−p(x,y)∗∏(x,y)∉E(1−p(x,y))
那么我们可以把
p(x,y)1−p(x,y)
作为边权,答案就是行列式*上面式子的右边。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=55;
const double eps=1e-8;
int n;
double a[N][N];
double gauss(int n)
{
double ans=1;
for (int i=1;i<=n;i++)
{
int l=i;
for (int j=i+1;j<=n;j++)
if (fabs(a[j][i])>fabs(a[l][i])) l=j;
if (l!=i) for (int j=i;j<=n;j++) swap(a[i][j],a[l][j]);
if (fabs(a[i][i])<eps) return 0;
for (int j=i+1;j<=n;j++)
{
if (fabs(a[j][i])<eps) continue;
double w=a[j][i]/a[i][i];
for (int k=i;k<=n;k++) a[j][k]-=a[i][k]*w;
}
ans*=a[i][i];
}
return fabs(ans);
}
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++)
scanf("%lf",&a[i][j]);
double tmp=1;
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++)
if (i!=j&&fabs(a[i][j])>eps)
{
double w=fabs(1-a[i][j])<eps?eps:1-a[i][j];
if (i<j) tmp*=w;
a[i][j]=-a[i][j]/w;
a[i][i]-=a[i][j];
}
printf("%.8lf",gauss(n-1)*tmp);
return 0;
}