PAT甲级 A1004 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)