BZOJ 1415: [Noi2005]聪聪和可可(记忆化搜索+期望)

传送门

解题思路

　　还是比较简答的一道题。首先\(bfs\)把每个点到其他点的最短路求出来，然后再记忆化搜索。记搜的时候猫的走法是确定的，搜一下老鼠走法就行了。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<queue>

using namespace std;
const int MAXN = 1005;

inline int rd(){
	int x=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {f=ch=='-'?0:1;ch=getchar();}
	while(isdigit(ch))  {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return f?x:-x;
}

int n,m,head[MAXN],cnt,dis[MAXN][MAXN],to[MAXN<<1],nxt[MAXN<<1],S,T,du[MAXN];
double ans,f[MAXN][MAXN];
bool vis[MAXN];
queue<int> Q;

inline void add(int bg,int ed){
	to[++cnt]=ed,nxt[cnt]=head[bg],head[bg]=cnt;
}

inline void bfs(int id){
	memset(vis,false,sizeof(vis));
	dis[id][id]=0;Q.push(id);vis[id]=1;int x,u;
	while(Q.size()){
		x=Q.front();Q.pop();
		for(int i=head[x];i;i=nxt[i]){
			u=to[i];if(vis[u]) continue;
			dis[id][u]=dis[id][x]+1;vis[u]=1;
			Q.push(u);
		}
	}
}

double dfs(int cat,int mouse){
	if(f[cat][mouse]!=0.0) return f[cat][mouse];
	int p=0;double sum=0;int prec=cat;
	for(int t=0;t<=1;t++){
		for(int i=head[cat];i;i=nxt[i]){
			int u=to[i];if(dis[mouse][u]>=dis[mouse][cat]) continue;
			if(!p || p>u) p=u;	
		}
		if(p==mouse) return 1.0;cat=p;p=0;
	}
	if(cat==mouse) return 1.0;
	for(int i=head[mouse];i;i=nxt[i]){
		int u=to[i];
		if(u==cat) sum+=1.0/(double)(du[mouse]+1);
		else sum+=(double)(dfs(cat,u)+1.0)/(double)(du[mouse]+1);
	}sum+=(double)(dfs(cat,mouse)+1.0)/(double)(du[mouse]+1);
	return f[prec][mouse]=sum;
}

int main(){
	memset(dis,0x3f,sizeof(dis));
	n=rd(),m=rd();int x,y;S=rd();T=rd();
	for(int i=1;i<=m;i++){
		x=rd(),y=rd();
		add(x,y),add(y,x);du[x]++;du[y]++;
	}
	for(int i=1;i<=n;i++) bfs(i);
	printf("%.3lf",dfs(S,T));
	return 0;
}