计蒜客家谱（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

代码：

#include<bits/stdc++.h>
using namespace std;
int n;
vector<int> son[100005];
int ans[100005];
bool vis[100005];
int dfs(int x){//dfs函数返回x节点的直系后代数
    int res=0;
    for(int i=0;i<son[x].size();i++){
        res+=dfs(son[x][i]);//遍历x节点的所有儿子，求出所有的儿子的直系后代数
    }
    ans[x]=res;
    return res+1;//x的孩子的直系后代数+1
}
int main(){
    cin>>n;
    for(int i=0;i<n-1;i++){
        int x,y;
        cin>>x>>y;
        son[x].push_back(y);//x的孩子为y
        vis[y]=true;//y有父亲，标记为true，表示不是根节点
    }
    int u;
    for(int i=1;i<=n;i++){//找根节点
        if(!vis[i]){
            u=i;
            break;
        }
    }
    dfs(u);
    for(int i=1;i<=n;i++){
        cout<<ans[i]<<endl;
    }
    return 0;
}