【NOI2005】【概率动态规划】聪聪和可可

这是一道求数学期望的题目，可以使用动态规划求解。

首先聪聪要逐步向可可靠近，所以我们按照题目要求预处理出p[i][j]表示i -> j的最短路上与i相邻且标号最小的点，可以使用n次spfa来实现。

聪聪下一步所在顶点为p[p[i][j]][j]，可可下一步可能在相邻的顶点或者不动,用w[j][i]表示

设计状态:f[i][j]表示聪聪在顶点i，可可在顶点j时聪聪抓住可可的平均步数

转移方程:

边界:

1、当i == j时，f[i][i] = 0;

2、当p[i][j] = j 或者 p[p[i][j]][j] = j时，f[i][j] = 1

这样我们就可以使用记忆化搜索解决这道题。

代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 1000 + 10;
const int maxque = 100000 + 10;
const int eps = 1e-12;
struct pnode
{
	int d;
	pnode *next;
	pnode(){}
	pnode(int d,pnode *next):d(d),next(next){}
}*first[maxn],__[maxn],*tot = __;
double f[maxn][maxn];
int path[maxn][maxn],deg[maxn],dis[maxn];
int que[maxque];
bool vis[maxn];
int n,m,s,t;

void init()
{
	freopen("bzoj1415.in","r",stdin);
	freopen("bzoj1415.out","w",stdout);
}

void spfa(int s)
{
	memset(que,0,sizeof(que));
	memset(vis,false,sizeof(vis));
	memset(dis,0x3f,sizeof(dis));
	int l = 0,r = 0;
	dis[s] = 0;
	que[r++] = s;vis[s] = true;
	while(l < r)
	{
		int u = que[l++];l %= maxque;
		vis[u] = false;
		for(pnode *p = first[u];p != NULL;p = p -> next)
		{
			int v = p -> d;
			if(dis[u] + 1 < dis[v] || (dis[u] + 1 == dis[v] && u < path[v][s]))
			{
				dis[v] = dis[u] + 1,path[v][s] = u;
				if(!vis[v])
				{
					que[r++] = v;
					vis[v] = true;r %= maxque;
				}
			}
		}
	}
}

void dp(int s,int t)
{
	if(s == t)
	{
		f[s][t] = 0;
		return;
	}
	if(f[s][t] > eps)return;
	int nxt = path[s][t];
	if(nxt == t)
	{
		f[s][t] = 1;
		return;
	}
	nxt = path[nxt][t];
	if(nxt == t)
	{
		f[s][t] = 1;
		return;
	}
	double ths = 0;
	for(pnode *p = first[t];p != NULL;p = p -> next)
	{
		dp(nxt,p -> d);
		ths += f[nxt][p -> d] / (deg[t] + 1);
	}
	dp(nxt,t);ths += f[nxt][t] / (deg[t] + 1);
	f[s][t] = ths + 1;
}

void solve()
{
	for(int i = 1;i <= n;i++)
		spfa(i);
	dp(s,t);
	printf("%.3lf\n",f[s][t]);
}

void readdata()
{
	scanf("%d%d",&n,&m);
	scanf("%d%d",&s,&t);
	for(int i = 1;i <= m;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		first[u] = new(tot++)pnode(v,first[u]);++deg[u];
		first[v] = new(tot++)pnode(u,first[v]);++deg[v];
	}
}

int main()
{
	init();
	readdata();
	solve();
	return 0;
}