[BZOJ3534][SDOI2014]重建（矩阵树定理）

最新推荐文章于 2020-07-20 10:27:53 发布

wwyx2001

最新推荐文章于 2020-07-20 10:27:53 发布

阅读量308

点赞数 1

分类专栏：矩阵树定理省选文章标签：矩阵树定理

本文链接：https://blog.csdn.net/blue_cuso4/article/details/80583344

版权

省选同时被 2 个专栏收录

74 篇文章 0 订阅

订阅专栏

矩阵树定理

6 篇文章 0 订阅

订阅专栏

题目：

我是超链接

题解：

矩阵树定理，先前只知道构造方法和基本的求生成树个数
然而实际上K矩阵是这么个玩意
这里写图片描述
我们可以发现其实是跟边权有关的
其实矩阵树定理求的是

\sum T \prod e \in T w e

$\sum _T\prod_{e\in T}w_e$
然后这个题目让我们求的显然是

\sum T \prod e \in T p e \prod e \notin T (1 - p e)

$\sum _T\prod_{e\in T}p_e\prod_{e\notin T}(1-p_e)$
这个式子求的前半截是一样的，后半截不太一样，但是显然

\prod e \notin T (1 - p e) = \prod e ( 1 - p e ) \prod e \in T ( 1 - p e )

$\prod_{e\notin T}(1-p_e)={\prod_{e}(1-p_e)\over \prod_{e\in T}(1-p_e)}$
然后就可以带进去了

\prod e (1 - p e) \sum T \prod e \in T p e 1 - p e

$\prod _e (1-p_e)\sum _T\prod _{e\in T}{p_e\over 1-p_e}$
然后我们就可以让a的权值为后面的一堆就好啦
还有就是当矩阵中出现

|a|<eps | a | < e p s $|a|<eps$ 时就

a=eps a = e p s $a=eps$ 当矩阵中出现

|1−a|<eps | 1 − a | < e p s $|1-a|<eps$ 时就

a=1−eps a = 1 − e p s $a=1-eps$
BZOJ上的输出一定要开到1e-10才能过= =

代码：

#include <cstdio>
#include <cmath>
#include <iostream>
using namespace std;
const int N=55;
const double eps=1e-8;
double a[N][N];
double gauss(int n)
{
    double ans=1;
    for (int i=1;i<=n;i++)
    {
        int num=i;
        for (int j=i+1;j<=n;j++)
          if (abs(a[num][i])<abs(a[j][i])) num=j;
        if (num!=i) for (int j=i;j<=n;j++) swap(a[num][j],a[i][j]);
        for (int j=i+1;j<=n;j++)
        {
            double t=a[j][i]/a[i][i];
            for (int k=i;k<=n;k++) a[j][k]-=t*a[i][k];
        }
        if (abs(a[i][i])<eps) return 0;
        ans=ans*a[i][i];
    }
    return abs(ans);
}
int main()
{
    int n;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 (abs(a[i][j])<eps) a[i][j]=eps;
        else if (abs(1.0-a[i][j])<eps) a[i][j]=1.0-eps;
        if (i<j) tmp*=1.0-a[i][j];
        a[i][j]=a[i][j]/(1.0-a[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];
    }
    printf("%.10lf",tmp*gauss(n-1));
}