PAT甲级真题1004. Counting Leaves

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

题意，给我们一棵家族树，给出这棵树的每一条边，问我们每一层没有子节点的节点数。

题意是非常的清晰，我们只需要在建立好的树上从根节点(01)开始遍历整棵树，在遍历的时候记录当前的层数即可。

#include <queue>
#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 100+5;
vector<int> tree[maxn];
int level[maxn], sum[maxn], Max;
int main() {
  int n, m;
  scanf("%d%d", &n, &m);
  level[1] = sum[1] = 0;
  for(int i=0; i<m; i++) {
    int x, num;
    scanf("%d%d", &x, &num);
    for(int j=0; j<num; j++) {
      int leaf;
      scanf("%d", &leaf);
      tree[x].push_back(leaf);
    }
  }
  Max = 0;
  queue<int> q;
  q.push(1);
  while(!q.empty()) {
    int x = q.front();
    q.pop();
    if(tree[x].size() == 0) {
      sum[level[x]]++;
      continue;
    }
    for(int i=0; i<tree[x].size(); i++) {
      int y = tree[x][i];
      level[y] = level[x] + 1;
      Max = max(Max, level[y]);
      q.push(y);
    }
  }
//  printf("%d\n", Max);
  for(int i=0; i<=Max; i++) {
    printf("%d", sum[i]);
    if(i != Max) {
      printf(" ");
    }
    else {
      printf("\n");
    }
  }
}