[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)即视为正确。

 

输入输出样例

输入样例#1： 复制

3
0 0.5 0.5
0.5 0 0.5
0.5 0.5 0

输出样例#1： 复制

0.375

说明

1 < N < =50

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

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

一棵生成树的概率就是所有存在的边的存在概率乘不存在的边的不存在概率

我们把每个边权设为$\frac{p(i,j)}{1-p(i,j)}$

然后求出生成树概率后乘以所有$1-p(i,j)$

如果没有选的边就会乘1-p(i,j)

如果有选的边就等价于

$\frac{p(i,j)}{1-p(i,j)}*(1-p(i,j))$

$p(i,j)$

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int n;
double eps=1e-9;
double sum,ans,a[51][51];
void guass()
{
    int i,j,now,k;
    n--;
    ans=1;
    for (i=1; i<=n; i++)
    {
        now=i;
        for (j=i+1; j<=n; j++)
        {
            if (fabs(a[j][i])>fabs(a[now][i])) now=j;
        }
        if (now!=i)
            for (j=i; j<=n; j++)
                swap(a[i][j],a[now][j]),ans=-ans;
        for (j=i+1; j<=n; j++)
        {
            double t=a[j][i]/a[i][i];
            for (k=i; k<=n; k++)
            {
                a[j][k]-=t*a[i][k];
            }
        }
    }
    for (i=1; i<=n; i++)
        ans=ans*a[i][i];
    ans=fabs(ans);
}
int main()
{
    int i,j;
    cin>>n;
    sum=1;
    for (i=1; i<=n; i++)
    {
        for (j=1; j<=n; j++)
        {
            scanf("%lf",&a[i][j]);
            if (i==j) continue;
            double tmp=1-a[i][j];
            if (tmp<=eps) tmp=eps;
            if (i<j)
                sum*=tmp;
            a[i][j]/=tmp;
        }
    }
    for (i=1; i<=n; i++)
    {
        for (j=1; j<=n; j++)
            if (j!=i)
            {
                a[i][i]+=a[i][j];
                a[i][j]=-a[i][j];
            }
    }
    guass();
    ans=sum*ans;
    printf("%.10lf\n",ans);
}