题目描述 Description
随着新版百度空间的上线,Blog宠物绿豆蛙完成了它的使命,去寻找它新的归宿。
给出一个有向无环图,起点为1终点为N,每条边都有一个长度,并且从起点出发能够到达所有的点,所有的点也都能够到达终点。绿豆蛙从起点出发,走向终点。
到达每一个顶点时,如果有K条离开该点的道路,绿豆蛙可以选择任意一条道路离开该点,并且走向每条路的概率为 1/K 。
现在绿豆蛙想知道,从起点走到终点的所经过的路径总长度期望是多少?
输入描述 Input Description
第一行: 两个整数 N M,代表图中有N个点、M条边
第二行到第 1+M 行: 每行3个整数 a b c,代表从a到b有一条长度为c的有向边
输出描述 Output Description
从起点到终点路径总长度的期望值,四舍五入保留两位小数。
样例输入 Sample Input
4 4
1 2 1
1 3 2
2 3 3
3 4 4
样例输出 Sample Output
7.00
数据范围及提示 Data Size & Hint
对于20%的数据 N<=100
对于40%的数据 N<=1000
对于60%的数据 N<=10000
对于100%的数据 N<=100000,M<=2*N
来源:Nescafe 19
DP。一开始我定义的dp[u]表示起点到u的期望,然后我发现这样计算很麻烦。
正解是定义dp[u]表示u到终点的期望,则可定义状态转移方程:
dp[v]+=(dp[u]+l[i].d)/cd[v]
其中边由v到u,cd[v]表示v的出度。
这个状态转移方程可以这样理解:
dp[v]+=dp[u]/cd[v]+l[i].d/cd[v]
dp[u]和l[i].d的概率都是1/cd[v],所以造成的影响的期望应该加起来。
因为是一个点可以由多条出边,所以要'+='
因为要无后效性,所以要拓扑排序。处理一个点之前,把它所有连向的点都处理完,所以拓扑排序要倒着来(我是直接反向建图)。
代码:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
const int size=1000010;
int head[size],nxt[size],tot=0;
struct edge{
int t;
double d;
}l[size];
void build(int f,int t,double d)
{
l[++tot].t=t;
l[tot].d=d;
nxt[tot]=head[f];
head[f]=tot;
}
double dp[size];
int n,m;
int cd1[size],cd2[size];
queue<int> q;
void toposort()
{
for(int i=1;i<=n;i++)
{
if(!cd1[i]) q.push(i);
}
while(q.size())
{
int x=q.front(); q.pop();
for(int i=head[x];i;i=nxt[i])
{
int v=l[i].t;
dp[v]+=(dp[x]+l[i].d)*1.0/cd2[v];
// printf("%d %d\n",x,v);
if(!--cd1[v]) q.push(v);
}
}
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
build(b,a,c);
cd1[a]++;
cd2[a]++;
}
toposort();
printf("%.2lf",dp[1]);
return 0;
}