两种做法:
1、使用队列进行BFS,层序遍历,每层开始时设置一个size记录当前层的结点数,每遍历完一层就输出一层的结果,最后一层结束后队列为空作为结束条件和格式输出条件
2、设置一个下标表示深度的数组,进行dfs,每遇到一个叶子结点就在对应深度加1,计算最大深度是为了最后的输出
Code1:
#include<iostream>
#include<vector>
#include<queue>
using namespace std;
vector<int> node[101];
queue<int> q;
int main() {
int n, m;
cin >> n >> m;
for (int i = 0; i < m; i++)
{
int p, num, c;
cin >> p >> num;
for (int j = 0; j < num; j++)
{
cin >> c;
node[p].push_back(c);
}
}
q.push(1);
while (!q.empty())
{
int size = q.size();
int cnt = 0;
for (int i = 0; i < size; i++)
{
int parent = q.front();
q.pop();
if (node[parent].empty()) cnt++;
else {
for (int j = 0; j < node[parent].size(); j++)
{
q.push(node[parent][j]);
}
}
}
if (q.empty()) {
cout << cnt;
}
else {
cout << cnt << " ";
}
}
return 0;
}
Code2:
#include<iostream>
#include<vector>
using namespace std;
vector<int> v[100];
int num[100];//存储每层的叶子节点个数
int max_depth = -1;
void dfs(int node, int depth)
{
if (v[node].size() == 0)
{
num[depth]++;
max_depth = max_depth > depth ? max_depth : depth;
return;
}
int cnt = v[node].size();
for (int i = 0; i < cnt; i++)
{
dfs(v[node][i], depth + 1);
}
}
int main()
{
int N;//总结点数
int M;//非叶子节点数
int k;//非叶子结点的孩子个数
int node;//非叶子结点编号
int c;//孩子节点编号
scanf("%d %d", &N, &M);
for (int i = 0; i < M; i++)
{
scanf("%d %d", &node, &k);
for (int j = 0; j < k; j++)
{
scanf("%d", &c);
v[node].push_back(c);
}
}
dfs(1,0);//root节点的编号为1,所在深度为0
printf("%d", num[0]);
for (int i = 1; i <= max_depth; i++)
{
printf(" %d", num[i]);
}
return 0;
}