题意:给出一颗树,每次删除一个节点及其子树(选每个点的概率都是相同的),直到删除了根节点。求步数的期望。
可以计算出选择每一个节点的概率,因为如果选择了一个节点,就在总的步数里加上一,所以把每个节点的概率加起来就是期望了。而概率就等于该节点的层数的倒数。
n = input()
edge = [[] for _ in range(n)]
for i in range(n - 1):
a, b = map(int, raw_input().split())
edge[a - 1].append(b - 1)
edge[b - 1].append(a - 1)
d = [0] * n
d[0] = 1
p = [0]
for u in p:
for v in edge[u]:
if not d[v]:
d[v] = d[u] + 1
p.append(v)
print sum((1. / x for x in d))
本文介绍了一种在树形结构中计算删除节点及其子树步数期望值的方法。通过计算每个节点被选中的概率,利用这些概率求和得到最终的期望值。文中提供了一个具体的Python代码实现。

被折叠的 条评论
为什么被折叠?



