[BZOJ 3036]绿豆蛙的归宿：期望DP

点击这里查看原题

f[i]表示从i到N的期望，因此f[i]为i能到的各个点的期望+边权的和除k。

/*
User:Small
Language:C++
Problem No.:3036
*/
#include<bits/stdc++.h>
#define ll long long
#define inf 999999999
using namespace std;
const int M=1e5+5;
double f[M];
int fir[M],n,m,tot,sum[M];
bool vis[M];
struct edge{
    int v,w,nex;
}e[M<<1];
void add(int u,int v,int w){
    e[++tot]=(edge){v,w,fir[u]};
    fir[u]=tot;
    sum[u]++;
}
void dfs(int u){
    if(vis[u]) return;
    vis[u]=1;
    for(int i=fir[u];i;i=e[i].nex){
        dfs(e[i].v);
        f[u]+=f[e[i].v]+e[i].w;
    }
    if(sum[u]) f[u]/=sum[u];
}
int main(){
    freopen("data.in","r",stdin);//
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
    }
    dfs(1);
    printf("%.2lf\n",f[1]);
    return 0;
}