PAT甲级 1004.Counting Leaves

参考：https://blog.csdn.net/qq278672818/article/details/54915636

首先贴上我一开始的部分正确代码：

 #include<bits/stdc++.h>
 using namespace std;
 const int N=1e4+;
 struct node
 {
     int level,child;//level为该节点层数，child为该节点孩子数
     node()
     {
         level=;
         child=;
     }
 }no[N];
 int n,m;
 int ans[N];//ans【i】为第i层叶子节点数
 int cmp(struct node x,struct node y)
 {
     return x.level<y.level;
 }
 int main()
 {
     cin>>n;
     if (n==)
         return ;
     cin>>m;
     no[].level=;
     int k,id,child,maxlevel=;//maxlevel为最大层数
     for (int i=;i<m;i++)
     {
         cin>>id>>k;
         no[id].child=k;
         for (int i=;i<k;i++)
         {
             cin>>child;
             no[child].level=no[id].level+;
             maxlevel=max(maxlevel,no[child].level);
         }
     }
     sort(no+,no++n,cmp);//按层数排序
     memset(ans,,sizeof(ans));
     for (int i=;i<=n;i++)
     {
         if (no[i].child==)
         {
             ans[no[i].level ]++;
         }
     }
     cout<<ans[];
     for (int i=;i<=maxlevel;i++)
         cout<<" "<<ans[i];
     cout<<endl;

     return ;
 }

经参考了上边的参考链接后发现：若有测试点是无序的，则该解法错误，因为节点的层数设置将不正确。

再贴上AC代码：

 #include<bits/stdc++.h>
 using namespace std;
 vector<int> ve[];
 int n,m,maxlevel=;//maxleve记录树的最大层数
 int ans[];//ans[i]保存第i层的叶子节点数
 void dfs(int level,int id)//level为当前层数，id为当前节点编号
 {
     if (ve[id].size()==)//找到叶子节点
     {
         maxlevel=max(maxlevel,level);
         ans[level]++;
         return;
     }
     for (int i=;i<ve[id].size();i++)//递归遍历孩子节点
     {
         dfs(level+,ve[id][i]);
     }
 }
 int main()
 {
     cin>>n;
     if (n==)
         return ;
     cin>>m;
     int id,k,child;
     for (int i=;i<m;i++)
     {
         cin>>id>>k;
         for (int j=;j<k;j++)
         {
             cin>>child;
             ve[id].push_back(child);
         }
     }
     memset(ans,,sizeof(ans));
     dfs(,);
     cout<<ans[];
     for (int i=;i<=maxlevel;i++)
     {
         cout<<" "<<ans[i];
     }
     cout<<endl;

     return ;
 }