PAT甲级-1004

传送门

题目大概意思是，求出一棵树中每一层叶子节点的数量，叶子节点即为没有任何子树的节点。我刚开始摸错门了，想着定义一个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;
}