1004 Counting Leaves (30point(s)) Easy only once

基本思想:

层序遍历问题;

 

关键点:

无;

 

#include
#include
#include
#include 
#include
#include
#include
#include
#include
#include
using namespace std;


int n, m;

struct node {
	vectorchild;
};

vectortree;
vectorseq;

/*
void layer_init() {
	if (tree.size() == 0)
		return;
	tree[0].layer = 0;
	queueq;
	q.push(0);
	while (!q.empty()){
		int index = q.front();
		q.pop();
		for (int i = 0; i < tree[index].child.size(); i++) {
			tree[tree[index].child[i]].layer = tree[index].layer + 1;
			q.push(tree[index].child[i]);
		}
	}
}
*/
	
void layer_find() {
	if (tree.size() == 0) {
		cout << 0 << endl;
		return;
	}
	queueq;
	q.push(1);
	while (!q.empty()){
		int size = q.size();
		int cnt = 0;
		for (int i = 0; i < size; i++) {
			int index = q.front();
			q.pop();
			if (tree[index].child.size() == 0)
				cnt++;
			else {
				for (int j = 0; j < tree[index].child.size(); j++) {
					q.push(tree[index].child[j]);
				}
			}
		}
		seq.push_back(cnt);
	}
}

int main() {
	cin >> n >> m;
	tree.resize(n+1);
	int a, b,c;
	for (int i = 0; i < m; i++) {
		cin >> a >> b;
		for (int j = 0; j < b; j++) {
			cin >> c;
			tree[a].child.push_back(c);
		}
	}
	layer_find();
	for (int i = 0; i < seq.size(); i++) {
		if (i == 0)
			cout << seq[i];
		else
			cout << " " << seq[i];
	}
	return 0;
}

  

你可能感兴趣的:(1004 Counting Leaves (30point(s)) Easy only once)