Robots 【计蒜客 ICPC南京网络赛】【期望DP】

题目链接

题目大意

有一个有向无环图，现在要从点1走到点n，第i天的花费为i，每天可以按照路走向相邻的饿节点或者保持不动，问走到点n的期望

解题思路

我们用天数期望去推花费期望
天数：me[u]=(1/(k+1)）* (me[v]+1) +(1/(k+1)）* (me[u]+1)
花费：dp[u]=(1/(k+1)）* (me[v]+dp[v]+1)+ (1/(k+1)）* (dp[u]+dp[v]+1)
其中k是和u相连的有多少个点，那么每次走向相邻点或者原地不动的概率就是(1/(k+1)）；
v是和u相邻的点
上面的公式化简一下就好
这个dp的顺序要用拓扑序才能保证计算的点不影响之前的点。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
const int N=1e5+5;
vector<int> rg[N];
int du[N],to[N],n,m;
int seq[N];
struct node
{
    int v,nex;
};
node e[N*4];
int f[N],tot;
void getDAGseq(int n)
{
    int top=0;
    queue<int> Q;
    for(int i=1;i<=n;++i){
        if(!du[i])
            Q.push(i);
    }
    while(!Q.empty())
    {
        int v=Q.front();
        Q.pop();
        seq[++top]=v;
        for(int u:rg[v]){
            du[u]--;
            if(du[u]==0)
                Q.push(u);
        }
    }
}
void add(int u,int v)
{
    e[tot].v=v;
    e[tot].nex=f[u];
    f[u]=tot++;
}
void init()
{
    memset(f,-1,sizeof(f));
    memset(to,0,sizeof(to));
    memset(du,0,sizeof(du));
    for(int i=0;i<=n;i++)
        rg[i].clear();
    tot=0;
}
double me[N],dp[N];
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        int x,y;
        init();
        for(int i=1;i<=m;i++)
        {
            scanf("%d %d",&x,&y);
            rg[y].push_back(x);
            du[x]++;
            add(x,y);
            to[x]++;
        }
        getDAGseq(n);//拓扑序
        memset(me,0,sizeof(me));
        memset(dp,0,sizeof(dp));
        for(int i=2;i<=n;i++)
        {
            int v=seq[i];
            double p=1.0/(double)(to[v]+1);
            for(int j=f[v];j!=-1;j=e[j].nex)
            {
                int vv=e[j].v;
                me[v]+=(me[vv]+1)*p;
            }
            me[v]+=p;
            me[v]*=1.0/(1.0-p);

            for(int j=f[v];j!=-1;j=e[j].nex)
            {
                int vv=e[j].v;
                dp[v]+=(me[vv]+dp[vv]+1)*p;
            }
            dp[v]+=(me[v]+1)*p;
            dp[v]*=1.0/(1.0-p);
            //printf("%f %f\n",me[v],dp[v]);
        }
        printf("%.2f\n",dp[1]);
    }
return 0;
}