The Largest Generation（DFS和BFS）

原题链接

PAT 甲级 1094

解题思路

这道题目要求每个深度下的结点个数，即每层的结点个数，然后输出最大的结点个数和其层数。

照样遍历树的过程中可以统计每一层的结点个数。

通过这几道关于树的遍历的DFS和BFS写法，可以发现DFS比BFS写起来要简洁，但是我觉得BFS更好理解。

源代码

DFS

DFS函数的参数为当前访问的结点下标index以及该结点所在的层数level。

#include<iostream>//DFS
#include<queue>
#include<vector>
#include<cstring>
using namespace std;
const int maxn=110;
vector<int> tree[maxn];
int level_num[maxn];//存放每个层次的结点个数
int n;
void DFS(int index, int level){
	level_num[level]++;
	for(int i=0; i<tree[index].size(); i++){
		int child = tree[index][i];
		DFS(child, level+1);
	}
}
int main()
{
    int m;
    scanf("%d%d",&n,&m);
    int father,cnt,child;
    for(int i=0; i<m; i++)
    {
        scanf("%d%d",&father,&cnt);
        for(int j=0; j<cnt; j++)
        {
            scanf("%d",&child);
            tree[father].push_back(child);
        }
    }
    memset(level_num,0,sizeof(level_num));
//    BFS(1);
	DFS(1, 1);//根结点为1号结点，层数为1 
    int res=1,generation=1;
    for(int i=2; i<n; i++)
    {
        if(level_num[i]>res)
        {
            res=level_num[i];
            generation=i;
        }
    }
    printf("%d %d\n",res,generation);
    return 0;
}

BFS

结点层数作为结点结构体的成员变量。

#include<iostream>//BFS
#include<queue>
#include<vector>
#include<cstring>
using namespace std;
const int maxn=110;
struct node
{
    int level;
    vector<int> child;
}tree[maxn];
int level_num[101];//存放每个层次的结点个数
int n;
void BFS(int root)
{
    queue<int> q;
    q.push(root);
    tree[root].level=1;//根节点是第一层
    while(!q.empty())
    {
//        for(int i=1; i<n; i++)
//            printf("%d ",level_num[i]);
//        printf("\n");
        int top=q.front();
        q.pop();
        level_num[tree[top].level]++;
        for(int i=0; i<tree[top].child.size(); i++)
        {
            int child=tree[top].child[i];
            tree[child].level=tree[top].level+1;
            q.push(child);
        }
    }
}
int main()
{
    int m;
    scanf("%d%d",&n,&m);
    int father,cnt,child;
    for(int i=0; i<m; i++)
    {
        scanf("%d%d",&father,&cnt);
        for(int j=0; j<cnt; j++)
        {
            scanf("%d",&child);
            tree[father].child.push_back(child);
        }
    }
    memset(level_num,0,sizeof(level_num));
    BFS(1);
    int res=1,generation=1;//注意要给初值，考虑只有根节点的情况
    for(int i=2; i<n; i++)
    {
        if(level_num[i]>res)
        {
            res=level_num[i];
            generation=i;
        }
    }
//    for(int i=1; i<n; i++)
//        printf("%d ",level_num[i]);
//    printf("\n");
    printf("%d %d\n",res,generation);
    return 0;
}