卡了两个测试点,一个估计是没有处理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;
}