洛谷 P4316 绿豆蛙的归宿

期望dp入门模板题。

大多数期望dp需要倒推，因为此时的期望值与后面的期望值相关。对于拓扑结构，只能用记忆化搜索来实现，对于线性结构，可以倒着dp，不过个人觉得记忆化搜索更容易理解也更容易代码实现。

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5,M=2e5+5;
int n,m,u,v;
double inf,w,du[N],f[N];
int cnt,head[N];
struct edge{int next,to; double w;}e[M]; 

inline void add(int u,int v,double w)
{
	cnt++;
	e[cnt].next=head[u];
	e[cnt].to=v;
	e[cnt].w=w;
	head[u]=cnt;
}

double dfs(int u)
{
	if (u==n) {f[u]=0; return f[u];}
	//终点到终点的期望为0
	if (f[u]!=inf) return f[u];
	f[u]=0;
	for (register int i=head[u]; i; i=e[i].next) f[u]+=dfs(e[i].to)+e[i].w;
	f[u]/=du[u];
	return f[u];
}

int main(){
	memset(f,-60,sizeof(f));
	inf=f[0];
	scanf("%d%d",&n,&m);
	for (register int i=1; i<=m; ++i) 
	{
		scanf("%d%d%lf",&u,&v,&w);
		add(u,v,w);
		du[u]=du[u]+1;
	}
	printf("%.2f\n",dfs(1));
return 0;	
}