BZOJ 3270 在各个房间相遇结束的概率

这题本来是上周五打cf前就想出来的。。。结果一直没写。。。

终于做到一种到一定精度就停下来的题了

然后不停得将概率推下去，直到概率小于某个精度就停下来，因为当t很大的时候，f[t][u][v]可能只有一点点了，不会对答案造成影响，结束就行。

之前eps定的1e-7WA了。感谢评论区提醒，开成1e-10就过了。

#include<bits/stdc++.h>
#define maxl 50001
#define eps 1e-10
using namespace std;

int n,m,cnt,a,b;
int ehead[21],du[21];
double p[21],ans[21];
double f[2][21][21];
struct ed
{
	int to,nxt;
}e[810];

inline void add(int u,int v)
{
	e[++cnt].to=v;e[cnt].nxt=ehead[u];ehead[u]=cnt;
}

inline void prework()
{
	scanf("%d%d%d%d",&n,&m,&a,&b);
	int u,v;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&u,&v);
		add(u,v);add(v,u);
		du[u]++;du[v]++;
	}
	for(int i=1;i<=n;i++)
	{	
		scanf("%lf",&p[i]);
		add(i,i);
	}
}

inline void mainwork()
{
	f[0][a][b]=1;
	int xx,yy;double tmp;
	for(int t=0;t<maxl-1;t++)
	{
		for(int u=1;u<=n;u++)
			for(int v=1;v<=n;v++)
				f[(t+1)&1][u][v]=0; 
		for(int u=1;u<=n;u++)
			for(int v=1;v<=n;v++)
			{
				if(u!=v)
				{
					if(f[t&1][u][v]<eps)
						continue;
					for(int i=ehead[u];i;i=e[i].nxt)
						for(int j=ehead[v];j;j=e[j].nxt)
						{
							if(e[i].to==u)
								tmp=p[u];
							else
								tmp=(1.0-p[u])/du[u];	
							if(e[j].to==v)
								tmp*=p[v];
							else
								tmp*=(1.0-p[v])/du[v];
							f[(t+1)&1][e[i].to][e[j].to]+=f[t&1][u][v]*tmp;
						}
				}
				else
					ans[u]+=f[t&1][u][u];
			}
	}
}

inline void print()
{
	for(int i=1;i<=n;i++)
		printf("%.6f ",ans[i]);
}

int main()
{
	prework();
	mainwork();
	print();
	return 0;
}