BZOJ 1415: [Noi2005]聪聪和可可|概率dp

论文题啊论文题可以参考《浅析竞赛中一类数学期望问题的解决方法》汤可因

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#include<set>
#include<map>
#define MX 1e9
#define lowbit(x) (x&(-x))
using namespace std;
int sc()
{
    int i=0;char c=getchar();
    while(c>'9'||c<'0')c=getchar();
    while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
    return i;
}
int dis[1001],vis[1001],from[1001],q[66666];
bool mark[1001][1001];
double f[1001][1001];
int head[1001],lst[2222],nxt[2222],D[2222];
int p[1001][1001];
int tot,n,m,S,W;
void insert(int x,int y)
{
    D[y]++;
    lst[++tot]=y;
    nxt[tot]=head[x];
    head[x]=tot;
}
void spfa(int S)
{
    for(int i=1;i<=n;i++)dis[i]=MX; dis[S]=0;
    int l=1,r=2; q[1]=S;
    while(l<r)
    {
        int x=q[l++]; vis[x]=0;
        for(int i=head[x];i;i=nxt[i])
        {
            if(dis[lst[i]]>dis[x]+1)
            {
                dis[lst[i]]=dis[x]+1;
                from[lst[i]]=x;
                if(!vis[lst[i]])vis[lst[i]]=1,q[r++]=lst[i];
            }
            else if(dis[lst[i]]==dis[x]+1&&from[lst[i]]>x)
                from[lst[i]]=x;
        }
    }
    for(int i=1;i<=n;i++)
        p[i][S]=from[i];
}
double dfs(int x,int y)
{
    if(mark[x][y])return f[x][y];
    mark[x][y]=1;
    if(x==y) return f[x][y]=0;
    if(y==p[x][y]||y==p[p[x][y]][y]) return f[x][y]=1;
    int t=p[p[x][y]][y];
    double ans=dfs(t,y);
    for(int i=head[y];i;i=nxt[i])
        ans+=dfs(t,lst[i]);
    return f[x][y]=ans/(D[y]+1)+1;
}
int main()
{
    n=sc(),m=sc(),S=sc(),W=sc();
    for(int i=1;i<=m;i++)
    {
        int x=sc(),y=sc();
        insert(x,y);
        insert(y,x);
    }
    for(int i=1;i<=n;i++)spfa(i);
    printf("%.3lf",dfs(S,W));
    return 0;
}