【CF446D】DZY Loves Games 高斯消元+矩阵乘法

【CF446D】DZY Loves Games

题意:一张n个点m条边的无向图,其中某些点是黑点,1号点一定不是黑点,n号点一定是黑点。问从1开始走,每次随机选择一个相邻的点走过去,经过恰好k个黑点到达n的概率。

$n\le 500,m\le 500000,k\le 10^9$,黑点个数不超过100.

题解:一眼就知道是高斯消元和矩乘什么的。我们先预处理出f[i][j]表示从第i个黑点开始走到的下一个黑点是j的概率。这个用高斯消元容易搞定。然后上矩乘即可。但是问题在于如果这样做的话我们要做n遍高斯消元。不过我们发现每次消元时左边的系数矩阵都是不变的,所以我们可以将n个方程组放到一起消元,复杂度就变成$O(n^3+10^6\log k)$了。

#include <cstring>
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
typedef double db;
int n,m,K,tot;
db f[510][610];
int dan[510],d[510],p[510],pa[100010],pb[100010];
struct M
{
	db v[105][105];
	M () {memset(v,0,sizeof(v));}
	db * operator [] (const int &a) {return v[a];}
	inline M operator * (const M &a) const
	{
		M b;
		int i,j,k;
		for(i=1;i<=tot;i++)	for(j=1;j<=tot;j++)	for(k=1;k<=tot;k++)	b.v[i][j]+=v[i][k]*a.v[k][j];
		return b;
	}
}S,T;
inline void pm(int y)
{
	while(y)
	{
		if(y&1)	S=S*T;
		T=T*T,y>>=1;
	}
}
inline int rd()
{
	int ret=0,f=1;	char gc=getchar();
	while(gc<'0'||gc>'9')	{if(gc=='-')	f=-f;	gc=getchar();}
	while(gc>='0'&&gc<='9')	ret=ret*10+(gc^'0'),gc=getchar();
	return ret*f;
}
int main()
{
	n=rd(),m=rd(),K=rd();
	int i,j,k,a,b;
	for(i=1;i<=n;i++)
	{
		dan[i]=rd();
		if(dan[i])	dan[i]=++tot,p[tot]=i;
	}
	for(i=1;i<=m;i++)	pa[i]=rd(),pb[i]=rd(),d[pa[i]]++,d[pb[i]]++;
	for(i=1;i<=m;i++)
	{
		a=pa[i],b=pb[i];
		if(!dan[a])	f[b][a]-=1.0/d[a];
		else	f[b][dan[a]+n]+=1.0/d[a];
		if(!dan[b])	f[a][b]-=1.0/d[b];
		else	f[a][dan[b]+n]+=1.0/d[b];
	}
	f[1][n+tot+1]=1;
	for(i=1;i<=n;i++)	f[i][i]+=1;
	for(i=1;i<=n;i++)
	{
		for(j=k=i;j<=n;j++)	if(fabs(f[j][i])>fabs(f[k][i]))	k=j;
		if(i!=k)	for(j=i;j<=n+tot+1;j++)	swap(f[i][j],f[k][j]);
		db tmp=f[i][i];
		for(j=i;j<=n+tot+1;j++)	f[i][j]/=tmp;
		for(j=1;j<=n;j++)	if(j!=i)
		{
			tmp=f[j][i];
			for(k=1;k<=n+tot+1;k++)	f[j][k]-=f[i][k]*tmp;
		}
	}
	for(i=1;i<=tot;i++)	for(j=1;j<=tot;j++)	T[i][j]=f[p[j]][i+n];
	for(i=1;i<=tot;i++)	S[1][i]=f[p[i]][n+tot+1];
	pm(K-2);
	printf("%.10lf",S[1][tot]);
	return 0;
}
posted @ 2018-04-05 16:53  CQzhangyu  阅读(643)  评论(0编辑  收藏  举报