题意:给出一颗树,每次删除一个节点及其子树(选每个点的概率都是相同的),直到删除了根节点。求步数的期望。
可以计算出选择每一个节点的概率,因为如果选择了一个节点,就在总的步数里加上一,所以把每个节点的概率加起来就是期望了。而概率就等于该节点的层数的倒数。
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))