我的PAT-ADVANCED代码仓:https://github.com/617076674/PAT-ADVANCED
原题链接:https://pintia.cn/problem-sets/994805342720868352/problems/994805521431773184
题目描述:
题目翻译:
1004 数叶子
家庭层级通常由谱系树呈现。 你的工作是统计那些没有孩子的家庭成员。
输入格式:
每个输入文件包含一个测试用例。在每个测试用例中,第一行给出了两个数字:0 < N < 100,代表树中的节点个数;M( < N)代表非叶子节点个数。接下来的M行,每行都是下述格式:
ID K ID[1] ID[2] ... ID[K]
ID是一个2位数字,表示一个非叶子节点,K是该节点的孩子数量,紧跟着的是一串ID值以及该ID值对应的孩子数量。为了简便,我们假设根节点的ID值是01。
输出格式:
对每个测试用例,从根结点开始,每一层你都需要统计那些没有孩子的成员。数字必须在1行内输出,每个数字间隔一个空格,行末不得有多余空格。
这个简单的例子只有2个节点,01代表根节点,02是它的唯一孩子。因此在和01同一层的节点中,没有叶子节点。而在下一层,有一个叶子节点。因此我们需要在1行内输出0 1。
输入样例:
2 1
01 1 02
输出样例:
0 1
知识点:树的层序遍历、树的深度优先遍历
思路一:树的层序遍历
本题的关键在于一棵普通树的保存。我们采取和PAT-ADVANCED1053——Path of Equal Weight同样的思路,用静态写法,事先开一个大小不低于节点上限个数的节点数组,用vector<int>型的变量来存储数组中的索引来代替指针域。
本解决思路和LeetCode102——二叉树的层序遍历思路一模一样,只不过LeetCode102——二叉树的层序遍历中的是一颗二叉树,而本题中的数并不一定是一颗二叉树。
在满足队列不为空条件的这个外循环里,我们先记录每一层的节点个数,即此时队列中存放的节点个数,记作qSize。每出队操作qSize个节点后的统计得到的叶子节点数是同一层的叶子节点数,这样就实现了分层计数的目的。将每层的计数结果保存在一个vector<int>型的变量leaves中即可。
时间复杂度和空间复杂度都是O(N)。
C++代码:
#include<iostream>
#include<vector>
#include<queue>
using namespace std;
struct node {
int num;
vector<int> child;
};
int N; //节点总数
int M; //非叶子节点数
node Node[100];
vector<int> leaves; //存放每一层的叶子节点数
void levelTraversal(int root);
int main(){
cin >> N >> M;
int ID, K, childID;
for(int i = 0; i < M; i++){
cin >> ID >> K;
for(int j = 0; j < K; j++){
cin >> childID;
Node[ID].child.push_back(childID);
}
}
levelTraversal(1);
for(int i = 0; i < leaves.size(); i++){
cout << leaves[i];
if(i != leaves.size() - 1){
cout << " ";
}
}
cout << endl;
return 0;
}
void levelTraversal(int root){
queue<int> q;
q.push(root);
while(!q.empty()){
int qSize = q.size();
int levelLeaves = 0;
for(int i = 0; i < qSize; i++){
int now = q.front();
q.pop();
if(Node[now].child.size() == 0){
levelLeaves++;
}
for(int j = 0; j < Node[now].child.size(); j++){
q.push(Node[now].child[j]);
}
}
leaves.push_back(levelLeaves);
}
}
C++解题报告:
思路二:树的深度优先遍历
另一个思路是使用树的深度优先遍历,在深度优先遍历的时候记录其层数,并按层数对每层进行叶子节点的计数操作。当然,输出的时候我们必须要知道这棵树的深度是多少,以免漏输出或者多输出0值。
时间复杂度和空间复杂度均是O(N)。
C++代码:
#include<iostream>
#include<vector>
#include<queue>
using namespace std;
struct node {
int num;
vector<int> child;
};
int N; //节点总数
int M; //非叶子节点数
node Node[100];
int leaves[100] = {0};//存放每一层的叶子节点数
int depth;
void dfs(int root, int level);
int main(){
cin >> N >> M;
int ID, K, childID;
for(int i = 0; i < M; i++){
cin >> ID >> K;
for(int j = 0; j < K; j++){
cin >> childID;
Node[ID].child.push_back(childID);
}
}
dfs(1, 0);
for(int i = 0; i <= depth; i++){
cout << leaves[i];
if(i != depth){
cout << " ";
}
}
cout << endl;
return 0;
}
void dfs(int root, int level){
if(Node[root].child.size() == 0){
leaves[level]++;
depth = max(depth, level);
return;
}
for(int i = 0; i < Node[root].child.size(); i++){
dfs(Node[root].child[i], level + 1);
}
}
C++解题报告: