1004. Counting Leaves (30)

卡了两个测试点，一个估计是没有处理0，另一个段错误，下次再写吧。。。。。

#include <iostream>
using namespace std;
#include <vector>
#include <algorithm>
#include <map>
struct Node{
    int val=0;
    int count=0;
    int node[4]={0};
    int root;
};
bool cmp(Node a,Node b){
    return a.root<b.root;
}
Node a[100];
int res[100] = {0};
map<int,int> Map;
int outCon[100] = {0};
int level =0;
void dfs(Node b){
    for(int i=0;i<b.count;i++){
        res[b.node[i]] = res[b.val]+1;
        if(res[b.node[i]] > level) level = res[b.node[i]] ;
        dfs(a[Map[b.node[i]]]);
    }
}
void dfs_leve(Node b){
    for(int i=0;i<b.count;i++){
        if(!Map[b.node[i]]){
            outCon[res[b.node[i]]]++;
        }
        dfs_leve(a[Map[b.node[i]]]);
    }
}
int main(){
    int n,m;
    cin>>n>>m;
    int tmp;
    for(int i=1;i<=m;i++){
        cin>>a[i].val>>a[i].count;
        Map[a[i].val] = i;
        if(a[i].val == 1) tmp = i;
        for(int j=0;j<a[i].count;j++){
            cin>>a[i].node[j];
        }
    }
    res[1] = 0;
    dfs(a[tmp]);
    dfs_leve(a[tmp]);
    for(int i=0;i<=level;i++){
        cout<<outCon[i];
        if(i!=level) cout<<" ";
    }
    return 0;
}