PAT甲级 1004.Counting Leaves
参考：https://blog.csdn.net/qq278672818/article/details/54915636
首先贴上我一开始的部分正确代码：
1 #include<bits/stdc++.h> 2 using namespace std; 3 const int N=1e4+5; 4 struct node 5 { 6 int level,child;//level为该节点层数，child为该节点孩子数 7 node() 8 { 9 level=10000; 10 child=0; 11 } 12 }no[N]; 13 int n,m; 14 int ans[N];//ans【i】为第i层叶子节点数 15 int cmp(struct node x,struct node y) 16 { 17 return x.level<y.level; 18 } 19 int main() 20 { 21 cin>>n; 22 if (n==0) 23 return 0; 24 cin>>m; 25 no[1].level=1; 26 int k,id,child,maxlevel=1;//maxlevel为最大层数 27 for (int i=0;i<m;i++) 28 { 29 cin>>id>>k; 30 no[id].child=k; 31 for (int i=0;i<k;i++) 32 { 33 cin>>child; 34 no[child].level=no[id].level+1; 35 maxlevel=max(maxlevel,no[child].level); 36 } 37 } 38 sort(no+1,no+1+n,cmp);//按层数排序 39 memset(ans,0,sizeof(ans)); 40 for (int i=1;i<=n;i++) 41 { 42 if (no[i].child==0) 43 { 44 ans[no[i].level ]++; 45 } 46 } 47 cout<<ans[1]; 48 for (int i=2;i<=maxlevel;i++) 49 cout<<" "<<ans[i]; 50 cout<<endl; 51 52 return 0; 53 }
经参考了上边的参考链接后发现：若有测试点是无序的，则该解法错误，因为节点的层数设置将不正确。
再贴上AC代码：
1 #include<bits/stdc++.h> 2 using namespace std; 3 vector<int> ve[110]; 4 int n,m,maxlevel=0;//maxleve记录树的最大层数 5 int ans[110];//ans[i]保存第i层的叶子节点数 6 void dfs(int level,int id)//level为当前层数，id为当前节点编号 7 { 8 if (ve[id].size()==0)//找到叶子节点 9 { 10 maxlevel=max(maxlevel,level); 11 ans[level]++; 12 return; 13 } 14 for (int i=0;i<ve[id].size();i++)//递归遍历孩子节点 15 { 16 dfs(level+1,ve[id][i]); 17 } 18 } 19 int main() 20 { 21 cin>>n; 22 if (n==0) 23 return 0; 24 cin>>m; 25 int id,k,child; 26 for (int i=0;i<m;i++) 27 { 28 cin>>id>>k; 29 for (int j=0;j<k;j++) 30 { 31 cin>>child; 32 ve[id].push_back(child); 33 } 34 } 35 memset(ans,0,sizeof(ans)); 36 dfs(0,1); 37 cout<<ans[0]; 38 for (int i=1;i<=maxlevel;i++) 39 { 40 cout<<" "<<ans[i]; 41 } 42 cout<<endl; 43 44 return 0; 45 }
