A1004 Counting Leaves (30分)

Posted 2021-03-09 tsruixi

一、技术总结

1. 这一题问题出现在答案不匹配，我开始的想法是在遍历的时候对于该层进行判断加加，但是总是有一个测试点过不去，不知道为啥。
2. 还是常规操作，就是在递归边界处，进行判断，然后加加，同时输出的时候需要注意，是深度同max_h是相等的，因为有这么多层。
3. 还有一些细节就是输出格式。

二、参考代码

#include<bits/stdc++.h>;
using namespace std;
const int maxn = 110;
struct node{
    vector<int> child;
}Node[maxn];
int hashTable[maxn] = {0};
int max_h = -1;
void DFS(int root, int depth){
    if(Node[root].child.size() == 0){
        hashTable[depth]++;
        max_h = max(depth, max_h);
        return;
    }
    for(int i = 0; i < Node[root].child.size(); i++){
        DFS(Node[root].child[i], depth+1);
    }   
}
int main(){
    int n, num;
    scanf("%d", &n);
    if(n == 0) return 0;
    scanf("%d", &num);
    int id, sum;
    for(int i = 0; i < num; i++){
        scanf("%d%d", &id, &sum);
        int id2;
        for(int j = 0; j < sum; j++){
            scanf("%d", &id2);
            Node[id].child.push_back(id2);
        }
    }
    DFS(1, 0);
    for(int i = 0; i <= max_h; i++){
        if(i != 0) printf(" ");
        printf("%d", hashTable[i]); 
    }
    return 0;
}