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

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

dfs遍历树

#include<bits/stdc++.h>

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

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[150];
int n,m;
int node ,k;
int vis[150];
int maxn=-1;
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=0;i<p[u.data].size();i++){
            Q.push(NODE(p[u.data][i],u.step+1));
        }
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>m;
    for(int i=0;i<m;i++){
        cin>>node>>k;
        int t;
        for(int j=0;j<k;j++){
            cin>>t;
            p[node].push_back(t);
        }
    }
    bfs(1,1);
    cout<<vis[1];
    for(int i=2;i<=maxn;i++){
        cout<<" "<<vis[i];
    }
    cout<<endl;
    return 0;
}