BZOJ 1415: [Noi2005]聪聪和可可

数学期望水题

预处理一下再记忆化搜索

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int N=1000+5;
const int inf=1e9;
struct Edge{int to,next;}e[N<<1];
int head[N],cnt;
void ins(int u,int v){
	e[++cnt]=(Edge){v,head[u]};head[u]=cnt;
}
int d[N][N];
void bfs(int s){
	memset(d[s],-1,sizeof(d[s]));
	queue<int>q;q.push(s);d[s][s]=0;
	while(!q.empty()){
		int u=q.front();q.pop();
		for(int i=head[u];i;i=e[i].next){
			int v=e[i].to;
			if(d[s][v]==-1){
				d[s][v]=d[s][u]+1;
				q.push(v);
			}
		}
	}
}
int p[N][N];
void calc(int s,int t){
	int ans=inf;
	for(int i=head[s];i;i=e[i].next){
		int v=e[i].to;
		if(d[v][t]<ans)ans=d[v][t],p[s][t]=v;
		else if(d[v][t]==ans)p[s][t]=min(p[s][t],v);
	}
}
double f[N][N],deg[N];
bool vis[N][N];
double dp(int s,int t){
	if(vis[s][t])return f[s][t];
	vis[s][t]=1;
	if(s==t)return f[s][t]=0.0;
	if(p[s][t]==t||p[p[s][t]][t]==t)return f[s][t]=1.0;
	f[s][t]=dp(p[p[s][t]][t],t);
	for(int i=head[t];i;i=e[i].next){
		int v=e[i].to;
		f[s][t]+=dp(p[p[s][t]][t],v);
	}
	return (f[s][t]/=(deg[t]+1.0))+=1.0;
}
int main(){
	//freopen("a.in","r",stdin);
	int n,m,s,t;
	scanf("%d%d%d%d",&n,&m,&s,&t);
	while(m--){
		int u,v;scanf("%d%d",&u,&v);
		ins(u,v);ins(v,u);deg[u]+=1.0;deg[v]+=1.0;
	}
	for(int i=1;i<=n;i++)bfs(i);
	for(int i=1;i<=n;i++)
	for(int j=1;j<=n;j++)
	if(i!=j)calc(i,j);
	printf("%.3lf\n",dp(s,t));
	return 0;
}