题目大概意思是,求出一棵树中每一层叶子节点的数量,叶子节点即为没有任何子树的节点。我刚开始摸错门了,想着定义一个tuple类型,后来又想建一棵树,不过后来发现思路太繁琐了。到了网上查了以后,发现BFS就可以解决问题了。
#include <iostream>
#include <vector>
#include <queue>
using std::string;
using std::cin;using std::cout;using std::vector;
using std::queue;
const int SIZE=100;
vector<vector<int>> v;
vector<int> BFS(int src){
queue<std::pair<int,int>> que;
vector<int> layer;
int cur_level=0,level=0;
int child_num=0;
que.push(std::make_pair(src,0));
//BFS
while(!que.empty()){
auto it=que.front();
cur_level=it.second;
if(level<cur_level){
level=cur_level;
layer.push_back(child_num);
child_num=0;
}
int index=it.first;
if(v[index].size()==0)
child_num++;
for(int i=0;i<v[index].size();i++)
que.push(std::make_pair(v[index][i],level+1));
que.pop();
}
layer.push_back(child_num);
return layer;
}
int main(int argc,char **argv){
int N,M,K;
string ID1,ID2;
v.resize(SIZE+1);
cin>>N>>M;
for(int i=0;i<M;i++){
cin>>ID1>>K;
for(int i=0;i<K;i++){
cin>>ID2;
v[std::stoi(ID1)].push_back(std::stoi(ID2));
}
}
auto it=BFS(1);
cout<<it[0];
for(int i=1;i<it.size();i++)
cout<<" "<<it[i];
return 0;
}