PAT-A 1004. Counting Leaves (30) (树)

题目链接：https://www.patest.cn/contests/pat-a-practise/1004

题意：

样例

输入：

2 1

01 1 02

输出：

0 1

第一行一共n个节点，其中非叶子节点有m个，下面m行，每行第一个数id是当前结点编号，第二个数k是当前结点有多少个孩子，后面k个数都是孩子的编号。

要求按顺序输出每一层的叶子节点的数量。

#include <iostream> 
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
struct node {
	int num; //该节点的孩子数 
	vector<int> child; //孩子的序号 
};
node T[110];
int ans[110], mh; //ans[i]表示第i层上有多少个叶子
//mh表示最大的层数 
void count_leaves(node rt, int h) { //参数rt表示当前节点，h表示当前结点层数 
	mh = mh < h ? h : mh; //不断维护层数的最大值 
	if(rt.num == 0) ans[h]++;   
	else {
		int i, j;
		for(i = 0; i < rt.child.size(); i++) {
			count_leaves(T[rt.child[i]], h + 1); //非叶子结点则递归判断 
		}
	}
}
int main() {
	int n, m;
	scanf("%d %d", &n, &m);
	int i, j, k, id, cd;
	for(i = 0; i < m; i++) {
		scanf("%d %d", &id, &k);
		T[id].num = k; // 以下标表示节点号 
		for(j = 0; j < k; j++) {
			scanf("%d", &cd);
			T[id].child.push_back(cd);
		}
	}
	count_leaves(T[1], 0);
	for(i = 0; i <= mh; i++) {
		if(i == 0) printf("%d", ans[i]);
		else printf(" %d", ans[i]);
	}
	printf("\n");
	return 0;
}