PAT 1094 树的BFS

题目

给定一棵树，求节点最多的一层的层号和节点个数，根节点为第一层

代码及思路

1. 任意树，所以用动态数组来建树，这样遍历的时候也很方便
2. 主要代码部分就是BFS，因为STL队列中提供了 size 函数，这样我们就可以很好的区分队列中的结点是不是一层的，而且也保证了可以每次只 pop 出本层结点的代号
3. 记录 level 可以在每次对当前所有的结点 pop 后，进行level++的操作
4. 对于最大层，只需要每次对比一下 maxsize 和当前的队列大小即可

#include<cstdio>
#include<vector>
#include<queue>
using namespace std;
const int maxn = 120;

vector<int> family[maxn];
int n, m, maxnum = -1;
int maxlevel;

void BFS(int s) {
	queue<int> q;
	int level = 1;
	q.push(s);
	while (!q.empty()) {
		int size = q.size();
		if (size > maxnum) { maxnum = size; maxlevel = level; }
		for (int i = 0; i < size; i++) {
			int key = q.front();
			q.pop();
			for (int j = 0; j < family[key].size(); j++) {
				int child = family[key][j];
				q.push(child);
			}
		}
		level++;
	}
	return;
}

int main() {
	scanf("%d %d", &n, &m);
	int id, child;
	int root;
	for (int i = 0; i < m; i++) {
		int numchild;
		scanf("%d %d", &id, &numchild);
		if (id == 1) root = i;
		for (int j = 0; j < numchild; j++) {
			scanf("%d", &child);
			family[id].push_back(child);
		}
	}
	BFS(1);
	printf("%d %d", maxnum, maxlevel);
}