1004 Counting Leaves （30 分）（树的遍历）

给出一棵树，问每一层各有多少叶子节点

dfs遍历树

#include<bits/stdc++.h>

using namespace std;
vector<int>p[];
int n,m;
int node ,k;
int vis[];
int maxn=-;
void dfs(int node,int step)
{
    if(p[node].empty()){
        vis[step]++;
        maxn=max(step,maxn);
    }
    else{
        for(int i=;i<p[node].size();i++){
            dfs(p[node][i],step+);
        }
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>m;
    for(int i=;i<m;i++){
        cin>>node>>k;
        int t;
        for(int j=;j<k;j++){
            cin>>t;
            p[node].push_back(t);
        }
    }
    dfs(,);
    cout<<vis[];
    for(int i=;i<=maxn;i++){
        cout<<" "<<vis[i];
    }
    cout<<endl;
    return ;
}

bfs遍历求树

 #include<bits/stdc++.h>

 using namespace std;
 struct NODE
 {
     int data;
     int step;
     NODE(){}
     NODE(int data1,int step1):data(data1),step(step1){}
 };
 vector<int>p[];
 int n,m;
 int node ,k;
 int vis[];
 int maxn=-;
 void bfs(int node,int step)
 {
     queue<NODE>Q;
     Q.push(NODE(node,step));
     while(!Q.empty()){
         NODE u=Q.front();
         Q.pop();
         if(p[u.data].empty()){
             maxn=max(u.step,maxn);
             vis[u.step]++;
         }
         for(int i=;i<p[u.data].size();i++){
             Q.push(NODE(p[u.data][i],u.step+));
         }
     }
 }
 int main()
 {
     ios::sync_with_stdio(false);
     cin>>n>>m;
     for(int i=;i<m;i++){
         cin>>node>>k;
         int t;
         for(int j=;j<k;j++){
             cin>>t;
             p[node].push_back(t);
         }
     }
     bfs(,);
     cout<<vis[];
     for(int i=;i<=maxn;i++){
         cout<<" "<<vis[i];
     }
     cout<<endl;
     return ;
 }