【随手记_PAT甲级踩坑记】1004 Counting Leaves

先上代码~
代码

#include<iostream>
#include<string.h>
using namespace std;
int main() {
	int n, m;
	cin >> n >> m;
	int* level = new int[n+1];//用来记录各个节点在树上的深度
	int* num_children = new int[n+1];//用来记录每个节点的孩子个数，同时也用来判断是否为叶子节点
	level[1] = 0;//把1的深度置为1
	for (int i = 0; i < n+1; ++i) {//初始化孩子个数
		num_children[i] = 0;
	}
	int** children = new int*[101];//一个指针数组，每个指针指向 -> 记录每个节点的孩子的数组
	for (int i = 0; i < m; ++i) {//存每个非叶节点的孩子、孩子个数
		int id,k;
		cin >> id>>k;
		num_children[id] = k;
		children[id] = new int[k];
		for (int j = 0; j < k; ++j) {
			cin >> children[id][j];
		}
	}
	int max_level = 0;//记录最大的深度
	for (int i = 1; i <= n; ++i) {
		if (num_children[i]!=0) {//
			for (int j = 0; j < num_children[i]; ++j) {//记录孩子节点的深度
				level[children[i][j]] = level[i] + 1;
			}
			if (level[i] + 1 > max_level)max_level = level[i] + 1;//最大深度更新
		}
	}
	int* num_leaf=new int[max_level+1];//每个深度的叶子个数
	memset(num_leaf, 0, (max_level + 1) * sizeof(int));
	for (int i = 1; i <= n; ++i) {
		if (num_children[i]==0) {//没有孩子就是叶子
			num_leaf[level[i]] += 1;//遍历每个节点，如果是叶子，对应的深度叶子个数+1
		}
	}
	for (int i = 0; i < max_level; ++i) {
		cout << num_leaf[i] << " ";
	}
	cout << num_leaf[max_level]<<endl;
	return 0;
}

• 测试点1和3的踩坑点：
  输入的时候不是按照顺序来的，也就是说第一行输入不是01开头的
  (本来想偷懒赌一把的，如果是按照顺序的话就可以直接在输入的时候计算深度，就不需要记录非叶节点的孩子了，结果赌输了hhh)