Codeforces 839C Journey(期望dp)

传送门

题意:给一棵大小为n,边长都为1的树,从1号点开始每次等概率地走向某个非父节点,走到叶节点为止,求走的长度的期望。

题解:

Sol 1. 定义f[x]为从x出发走到不能再走的期望,dfs一次,用子节点更新父节点。

Sol 2. 正推走到每个点的概率,把每个叶节点的深度(1号点为0)乘以走到它的概率加到ans里即可。

总之倒推期望,正推概率,突然发现正推概率精度不会丢哈哈

 

Sol 1:

#include
#include
#include
#include
using namespace std;
typedef long long ll;
const int N=1e5+4;
int n;
struct Edge {
	int v,nxt;
}e[N<<1];
int head[N],etot;
int son[N];
double f[N];
inline int read() {
	int x=0,f=1;char c=getchar();
	while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
	while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
	return x*f;
}
inline void adde(int u,int v) {
	e[++etot].nxt=head[u],e[etot].v=v,head[u]=etot;
}
inline void dfs(int p,int fa) {
	for (int i=head[p];~i;i=e[i].nxt) {
		int v=e[i].v;
		if (v^fa) {
			++son[p];
			dfs(v,p);
		}
	}
	for (int i=head[p];~i;i=e[i].nxt) {
		int v=e[i].v;
		if (v^fa) {
			f[p]+=(f[v]+1.0)/son[p];
		}
	}
}
int main() {
	memset(head,-1,sizeof(head));
	n=read();
	for (register int i=1;i

Sol 2:

#include
#include
#include
#include
using namespace std;
typedef long long ll;
const int N=1e5+4;
int n;
struct Edge {
	int v,nxt;
}e[N<<1];
int head[N],etot;
int son[N],dep[N];
double f[N];
inline int read() {
	int x=0,f=1;char c=getchar();
	while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
	while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
	return x*f;
}
inline void adde(int u,int v) {
	e[++etot].nxt=head[u],e[etot].v=v,head[u]=etot;
}
inline void dfs(int p,int fa) {
	dep[p]=dep[fa]+1;
	for (int i=head[p];~i;i=e[i].nxt) {
		int v=e[i].v;
		if (v^fa) ++son[p];
	}
	for (int i=head[p];~i;i=e[i].nxt) {
		int v=e[i].v;
		if (v^fa) {
			f[v]=f[p]/son[p];
			dfs(v,p);
		}
	}
}
int main() {
	memset(head,-1,sizeof(head));
	n=read();
	for (register int i=1;i

 

你可能感兴趣的:(期望dp,期望dp)