【DFS题型八/树状图】家谱

这一天蒜头君拿到了自己家的家谱，蒜头君便想知道，在自己家的家谱中，每位祖先有多少直系后代（直系后代包括他的孩子和他孩子的直系后代）。但是家族历史源远流长，家谱实在太庞大了，自己一个人完全数不过来。热心的你便自告奋勇帮蒜头君写一个程序，来统计每位祖先有多少直系后代。

输入格式
输入的第一行有一个整数 n(1≤n≤100000)，表示家谱中的总人数。

接下来读入 n−1 行，每行有两个整数 x(1≤x≤n), y(1≤y≤n)，表示 x 是 y 的父母。

输出格式
输出 n 行，每行有一个整数，表示第 i 个人有多少个直系后代。

样例输入

4

1 2

1 3

2 4

样例输出

3

1

0

0

n=int(input())
son=[[] for i in range(n)]
f=[0]*n
ans=[0]*n
for i in range(n-1):#建图
    a,b=map(int,input().split())
    son[a-1].append(b-1)
    f[b-1]=True#标记根节点，即无父母的，找到根节点
#print(son)
for i in range(n):#找到根节点
    if f[i]==False:
        u=i

def dfs(u):
    ret=0
    for i in son[u]:
        ret+=dfs(i)
    ans[u]=ret
    return ret+1

dfs(u)
for i in range(n):
    print(ans[i])