PAT A1094 The Largest Generation [树的遍历]

题目描述

链接
输入树的结点个数N，结点编号为1~N，非叶子结点个数M，然后输出M个非叶子结点的孩子结点的编号，求结点个数最多的一层，根结点的层号为1，输出该层的结点个数以及层号
这道题类似于前面的那个可以用DFS,BFS解的题

BFS

void bfs(){
    queue<int> q;
    layer = 1;
    q.push(1);
    while(!q.empty()){
        int len = q.size(); //因为后面push会变
        for(int i=0;i<len;i++){ //对于该层所有结点，每访问一个cnt[layer]++
            int now = q.front();
            q.pop();
            cnt[layer]++;
            for(int j=0;j<v[now].size();j++){
                q.push(v[now][j]);
            }
        }
        layer++; //在这里layer++
    }
}

DFS

太简洁了！！！

void dfs(int root, int layer) {
    cnt[layer]++; //每层cnt++
    for(int i = 0; i < v[root].size(); i++)
        dfs(v[root][i], layer+1);
}

完整代码

#include<bits/stdc++.h>
using namespace std;

const int maxn = 105;
vector<int> v[maxn];
int n,m,id,k,chid;
int layer;
int cnt[maxn];
void bfs(){
    queue<int> q;
    layer = 1;
    q.push(1);
    while(!q.empty()){
        int len = q.size();
        for(int i=0;i<len;i++){
            int now = q.front();
            q.pop();
            cnt[layer]++;
            for(int j=0;j<v[now].size();j++){
                q.push(v[now][j]);
            }
        }
        layer++;
    }
}

void solve(){
    int maxlevel = 1;
    for(int i=1;i<=layer;i++){
        if(cnt[maxlevel] < cnt[i]){
            maxlevel = i;
        }
    }
    printf("%d %d
", cnt[maxlevel], maxlevel);
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++){
        scanf("%d%d",&id,&k);
        for(int j=0;j<k;j++){
            scanf("%d",&chid);
            v[id].push_back(chid);
        }
    }
    bfs();
    solve();
}