BZOJ 3036: 绿豆蛙的归宿 期望dp

3036: 绿豆蛙的归宿

Time Limit: 2 Sec  Memory Limit: 128 MB
Submit: 520  Solved: 375
[Submit][Status][Discuss]

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

说好的刷水有益健康呢，为啥WA两发      哭

#include<ctime>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>
using namespace std;
typedef double db;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}
const int N=100100;
int ecnt,n,m,last[N],outdegree[N];
db dp[N];
bool book[N];
struct EDGE{int to,nt,val;}e[N<<1];
inline void add(int u,int v,db val)
{e[++ecnt]=(EDGE){v,last[u],val};last[u]=ecnt;}
db dfs(int u)
{
	if(book[u])return dp[u];book[u]=1;
	db k=outdegree[u];
	for(int i=last[u];i;i=e[i].nt)
	{dp[u]+=(dfs(e[i].to)+e[i].val)/k;}
	return dp[u];
}
void solve()
{printf("%.2lf\n",dfs(1));}
int main()
{
	n=read();m=read();int u,v,val;
	for(int i=1;i<=m;i++)
	{
		u=read();v=read();val=read();
		add(u,v,val);outdegree[u]++;
	}
	solve();
	return 0;
}
/*
4 4
1 2 1
1 3 2
2 3 3
3 4 4

7.00
*/