【NOI2005】聪聪与可可 题解(最短路+期望DP)

前言:学长讲的太神了;自己还能推出来DP式子,挺开心。

--------------------------

题目链接

题目大意:给定一张含有$n$个结点$m$条边的无向连通图。现在聪聪在点$s$,可可在点$t$。每秒钟可可能等概率走向相邻的结点或原地不动,而聪聪总是向更靠近可可的地方沿最短路走两步(如果走一步就能找到可可就不往下走了)。问聪聪找到可可的时间的期望。$n,m\leq 1000$

----------------------

我们首先解决第一个限制条件:沿最短路走。

假设聪聪目前在点$i$,可可目前在点$j$,聪聪下一步的走位是$next[i][j]$。

看到数据范围,我们可以暴力把每个点的单源最短路径求出来,然后枚举距离点$i$距离为$1$的点$k$。如果$dis[i][j]-1==dis[k][j]$,那么$next[i][j]=k$。

然后进行期望DP。这里我们采用记忆化搜索。设$f[i][j]$表示目前聪聪在点$i$,可可在点$j$时的期望。设点的出度为$du[]$。然后分类讨论:

  1.如果$i$和$j$同点,那么$f[i][j]=0$。

  2.如果聪聪能够走一步或两步到达点$j$,那么$f[i][j]=1$。

  3.如果可可呆在原地不动,那么对答案的贡献有$(f[next[next[i][j]][j]][j]+1)*\frac{1}{du[j]+1}$。(一共有$du[j]+1$种走法,包含原地不动)

  4.如果可可走向相邻的点,那么对答案的贡献有$\sum (f[next[next[i][j]][j]][to]+1)*\frac{1}{du[j]+1}$。(枚举$to$)

所以总的DP方程为$f[i][j]=\frac{f[next[next[i][j]][j]][j]+\sum f[next[next[i][j]][j]][to]}{du[j]+1}+1$

最后输出$dfs(s,t)$即可。时间复杂度$O(n^2)$。

代码:

#include<bits/stdc++.h>
using namespace std;
int dis[][],next[][],du[],vis[];
int n,m,s,t,visit[][];
double f[][];
int head[],cnt;
struct node
{
int next,to,dis;
}edge[];
inline int read()
{
int x=,f=;char ch=getchar();
while(!isdigit(ch)){if (ch=='-') f=-;ch=getchar();}
while(isdigit(ch)){x=x*+ch-'';ch=getchar();}
return x*f;
}
inline void add(int from,int to,int dis)
{
edge[++cnt].next=head[from];
edge[cnt].to=to;
edge[cnt].dis=dis;
head[from]=cnt;
}
inline void spfa(int x)
{
queue<int> q;
dis[x][x]=;vis[x]=;q.push(x);
while(!q.empty())
{
int now=q.front();q.pop();vis[now]=;
for (int i=head[now];i;i=edge[i].next)
{
int to=edge[i].to;
if (dis[x][to]>dis[x][now]+edge[i].dis)
{
dis[x][to]=dis[x][now]+edge[i].dis;
if (!vis[to]) q.push(to),vis[to]=;
}
}
}
}
double dfs(int u,int v)
{
if (visit[u][v]) return f[u][v];
if (u==v) return ;
int fir=next[u][v];
int sec=next[fir][v];
if (fir==v||sec==v) return ;
f[u][v]=;
for (int i=head[v];i;i=edge[i].next)
{
int to=edge[i].to;
f[u][v]+=dfs(sec,to)/(double)(du[v]+);
}
f[u][v]+=dfs(sec,v)/(double)(du[v]+);
visit[u][v]=;
return f[u][v];
}
int main()
{
n=read(),m=read(),s=read(),t=read();
for (int i=;i<=m;i++)
{
int x=read(),y=read();
add(x,y,);
add(y,x,);
du[x]++,du[y]++;
}
for (int i=;i<=n;i++)
for (int j=;j<=n;j++) dis[i][j]=next[i][j]=0x3f3f3f3f;
for (int i=;i<=n;i++) spfa(i);
for (int i=;i<=n;i++)
for (int j=head[i];j;j=edge[j].next)
{
int to=edge[j].to;
for (int k=;k<=n;k++)
if (dis[i][k]-==dis[to][k]) next[i][k]=min(next[i][k],to);
}
printf("%.3lf",dfs(s,t));
return ;
}
上一篇:php写一个简洁的登录页面


下一篇:HTML5第一讲