洛谷P4316题解

题目传送门：绿豆蛙的归宿 - 洛谷

---

这是一道非常明显的期望概率DP题目。

考虑到题目要求我们求路径总长度的期望值，我们可以考虑每条边对于答案的贡献，于是得到一个一个很显然的做法——用每条边的长度乘上经过他的概率，即：

此时问题就转化为了如何求p[i]。

我们设该边的起点为u，出度为k，假设我们知道从1出发到u的概率为dp[u]，显然：

那么怎么求dp[u]呢？我们可以通过其前继点的概率乘上他每条出边的概率求出，此时我们会发现一个问题，就是如果我们从1出发遍历这个图的话，可能出现一下这种情况：

我们在通过1-3-5这条路径求出5的概率后dp[3]又会被1-4-3这条路径更新，此时dp具有了后效性。为了避免这种情况的发生，我们可以建反图，从终点出发记忆化搜索，保证每个点在所有入边考虑完后再来更新后面的点。

代码实现

#include<bits/stdc++.h>
using namespace std;
int const N=1e5+5;
double p[N],dp[N];
struct edge{
	int u,v,w;
}e[N<<1];
int first[N],nxt[N<<1],n,m,out[N],cnt;
inline void add(int u,int v,int w){
	e[++cnt].u=u;e[cnt].v=v;e[cnt].w=w;
	nxt[cnt]=first[u];first[u]=cnt;
}
inline double dfs(int u){
	if(dp[u]!=0)return dp[u];
	for(int i=first[u];i;i=nxt[i]){
		int v=e[i].v;
		dp[u]+=dfs(v)*p[v];
	}
	return dp[u];
}
int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin>>n>>m;
	for(int i=1;i<=m;i++){
		int u,v,w;
		cin>>u>>v>>w;
		out[u]++;
		add(v,u,w);//建反图
	}
	for(int i=1;i<=n;i++){
		p[i]=1.0/out[i];
	}
	dp[1]=1;//初始化
	dp[n]=dfs(n);
	double ans=0;
	for(int i=1;i<=cnt;i++){
		ans+=e[i].w*dp[e[i].v]*p[e[i].v];
	}
	printf("%.2lf",ans);
	return 0;
}