【bzoj1415】 Noi2005—聪聪和可可

http://www.lydsy.com/JudgeOnline/problem.php?id=1415 (题目链接)

题意

　　一张图，聪聪想吃可可。每单位时间聪聪可以先移动两次；可可后移动一次或停在原地，它们的概率相等。问聪聪吃到可可的期望时间。

Solution

　　先bfs跑出两点间的最短距离。然后记忆化搜索就可以了，很水。。

代码

// bzoj1415
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
#define LL long long
#define inf 1<<30
#define Pi acos(-1.0)
#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
using namespace std;

const int maxn=1010;
struct edge {int to,next;}e[maxn<<1];
int head[maxn],r[maxn],S,T,n,m,cnt;
int dis[maxn][maxn],vis[maxn][maxn];
double f[maxn][maxn];

void link(int u,int v) {
	e[++cnt]=(edge){v,head[u]};head[u]=cnt;
	e[++cnt]=(edge){u,head[v]};head[v]=cnt;
}
void bfs(int s,int *d) {
	queue<int> q;q.push(s);
	while (!q.empty()) {
		int x=q.front();q.pop();
		for (int i=head[x];i;i=e[i].next)
			if (!d[e[i].to] && e[i].to!=s) d[e[i].to]=d[x]+1,q.push(e[i].to);
	}
}
double dfs(int s,int t) {
	if (vis[s][t]) return f[s][t];
	if (dis[s][t]==0) return 0;
	if (dis[s][t]<=2) return 1;
	vis[s][t]=f[s][t]=1;
	int k=inf,tmp=s;
	for (int i=head[tmp];i;i=e[i].next)
		if (dis[t][e[i].to]<dis[tmp][t] && e[i].to<k) k=e[i].to;
	tmp=k;k=inf;
	for (int i=head[tmp];i;i=e[i].next)
		if (dis[t][e[i].to]<dis[tmp][t] && e[i].to<k) k=e[i].to;
	tmp=k;
	for (int i=head[t];i;i=e[i].next)
		f[s][t]+=(double)1/(r[t]+1)*dfs(tmp,e[i].to);
	f[s][t]+=(double)1/(r[t]+1)*dfs(tmp,t);
	return f[s][t];
}
int main() {
	scanf("%d%d",&n,&m);
	scanf("%d%d",&S,&T);
	for (int u,v,i=1;i<=m;i++) {
		scanf("%d%d",&u,&v);
		link(u,v);r[u]++;r[v]++;
	}
	for (int i=1;i<=n;i++) bfs(i,dis[i]);
	printf("%.3lf",dfs(S,T));
	return 0;
}