[BZOJ3036]绿豆蛙的归宿（拓扑序+期望dp）

题目描述

传送门

题解

DAG上的sb dp

代码

#include
#include
#include
#include
#include
#include
using namespace std;
#define N 500005

int n,m,out[N];
int tot,point[N],nxt[N],v[N];double c[N];
int _tot,_point[N],_nxt[N],_v[N];
double d[N],f[N];
queue <int> q;

void add(int x,int y,double z)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
    ++_tot; _nxt[_tot]=_point[y]; _point[y]=_tot; _v[_tot]=x;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;++i)
    {
        int x,y,z;scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);d[x]+=1.0;++out[x];
    }
    q.push(n);
    while (!q.empty())
    {
        int x=q.front();q.pop();
        for (int i=point[x];i;i=nxt[i])
            f[x]+=(f[v[i]]+c[i])*(1/d[x]);
        for (int i=_point[x];i;i=_nxt[i])
        {
            --out[_v[i]];
            if (!out[_v[i]]) q.push(_v[i]);
        }
    }
    printf("%.2lf\n",f[1]);
}