bzoj 3534: [Sdoi2014]重建【矩阵树定理】

啊啊啊无脑背过果然不可取
比如这道题就不会写
参考:https://blog.csdn.net/iamzky/article/details/41317333

#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
const int N=55;
const double eps=1e-7;
int n;
double a[N][N],tmp=1;
int cmp(double x)
{
	return x<-eps?-1:x>eps;
}
double gaosi(int n)
{
	double ans=1;
	for(int i=1;i<=n;i++)
	{
		int nw=i;
		for(int j=i+1;j<=n;j++)
			if(fabs(a[j][i])>fabs(a[nw][i]))
				nw=j;
		for(int j=1;j<=n;j++)
			swap(a[nw][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]-=a[i][k]*t;
		}
		if(!cmp(a[i][i]))
			return 0;
		ans*=a[i][i];
	}
	return fabs(ans);
}
int main()
{
	scanf("%d\n",&n);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		{
			scanf("%lf",&a[i][j]);
			if(i!=j)
			{
				if(a[i][j]>1-eps)
					a[i][j]-=eps;
				if(i<j)
					tmp*=1.0-a[i][j];
				a[i][j]=a[i][j]/(1-a[i][j]);
			}
		}
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			if(i!=j)
				a[i][i]-=a[i][j];
	printf("%.8f\n",gaosi(n-1)*tmp);
	return 0;
}
posted @ 2018-07-03 16:41  lokiii  阅读(170)  评论(0编辑  收藏  举报