（ACWing217）绿豆蛙的归宿（数学期望）

最新推荐文章于 2022-03-27 22:28:22 发布

AC__dream

最新推荐文章于 2022-03-27 22:28:22 发布

阅读量149

点赞数 2

分类专栏：概率与期望文章标签：数学期望

本文链接：https://blog.csdn.net/AC__dream/article/details/120899849

版权

概率与期望专栏收录该内容

16 篇文章 0 订阅

订阅专栏

这篇博客探讨了如何解决一个数学期望问题，即在有向无环图（DAG）中，从起点到终点经过的路径长度的期望值。作者通过记忆化搜索的方法，从终点开始反向计算每个节点的路径期望，并给出了相应的C++代码实现。这个算法适用于起点唯一、终点不唯一的期望问题。

摘要由CSDN通过智能技术生成

题目：

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

对于数学期望的问题，一般起始点是唯一的，终点不一定是唯一的，我们可以从终点开始进行记忆化搜索，每个点的路径期望等于可以到达他的所有点的期望加上边权再乘以概率，下面是代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
using namespace std;
const int N=2e5+10;
int h[N],e[N],ne[N],w[N],idx,chu[N];
//dp[i]表示i点到达n点的路径长度的期望 
double f[N];
int n;
void add(int x,int y,int z)
{
	e[idx]=y;
	w[idx]=z;
	ne[idx]=h[x];
	h[x]=idx++;
}
double dp(int x)
{
	if(f[x]) return f[x];
	if(x==N) return 0;
	for(int i=h[x];i!=-1;i=ne[i])//每个点的期望都要考虑到所有能一步到达他的点的期望 
	{
		int j=e[i];
		f[x]+=1.0*(w[i]+dp(j))/chu[x];
	}
	return f[x];
}
int main()
{
	int m;
	scanf("%d%d",&n,&m);
	int a,b,c;
	memset(h,-1,sizeof h);
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&a,&b,&c);
		add(a,b,c);
		chu[a]++;
	}
	printf("%.2lf",dp(1));
	return 0; 
}