BZOJ 3036 绿豆蛙的归宿 期望DP

Time Limit: 2 Sec Memory Limit: 128 MB
Submit: 610 Solved: 431

Description

随着新版百度空间的下线，Blog宠物绿豆蛙完成了它的使命，去寻找它新的归宿。

给出一个有向无环的连通图，起点为1终点为N，每条边都有一个长度。绿豆蛙从起点出发，走向终点。
到达每一个顶点时，如果有K条离开该点的道路，绿豆蛙可以选择任意一条道路离开该点，并且走向每条路的概率为 1/K 。
现在绿豆蛙想知道，从起点走到终点的所经过的路径总长度期望是多少？

Input

第一行: 两个整数 N M，代表图中有N个点、M条边
第二行到第 1+M 行: 每行3个整数 a b c，代表从a到b有一条长度为c的有向边

Output

从起点到终点路径总长度的期望值，四舍五入保留两位小数。

Sample Input

4 4

1 2 1

1 3 2

2 3 3

3 4 4

Sample Output

7.00

HINT

对于100%的数据 N<=100000，M<=2*N

Source

Poetize3

---

题解：

搜索一下，因为是有向无环图，所以直接每次往下dfs，然后对于点u，我们dfs所有孩子v，并且每次加上v的dp值（从v到终点的期望长度）和u到v的长度，最后除以一个出度即可

---

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int MAXN=100000+10;
int out[MAXN],vis[MAXN],head[MAXN],tail,n,m;
double dp[MAXN];
struct Line{int to,nxt,flow;}line[MAXN*2];
void add_line(int from,int to,int flow){line[++tail].to=to;line[tail].nxt=head[from];head[from]=tail;line[tail].flow=flow;}
void dfs(int u){
    if(vis[u]) return;vis[u]=1;
    for(register int i=head[u];i;i=line[i].nxt){int v=line[i].to;dfs(v);dp[u]=dp[u]+line[i].flow+dp[v];}
    if(out[u])dp[u]/=out[u];
}
int main(){
    scanf("%d%d",&n,&m);
    for(register int i=1;i<=m;i++){int u,v,w;scanf("%d%d%d",&u,&v,&w);add_line(u,v,w);out[u]++; }
    dfs(1);
    printf("%.2lf\n",dp[1]);
    return 0;
}