pat-1004

题目

1. 题目

2. 测试用例

input example:

2 1
01 1 02

out example:

0 1

输入：每一个输入文件包含一个测试用例。每一个测试用例是一行包含了0-100个数据N，树的节点的个数和M，没有叶子的节点。
接下来的M行，每一个有以下的格式：
ID是两个字节数代表没有叶子的节点，K是他的孩子的数量，跟着的是孩子的ID
输出：对于每一个测试案例，你能够去数没有孩子的家庭成员
从根节点01 level来看，0个叶子结点；然后下一级有1个叶子姐点

一、思路

dfs,深度优先遍历。

关键
数据结构：
1.动态数组vector tree[100]
由于不清楚子节点的个数，所以利用动态数组
2.int num[100]
树的层数最多为100
判断条件：
出口条件：叶节点，同时更新最大深度 与 当前层叶节点数目；
依次遍历（调用dfs）该节点的每个叶节点；

二、代码（c++）

#include<iostream>
#include<vector>

using namespace std;

int num[100]={0};
vector<int>tree[100];
int maxdepth=1;

void dfs(int node,int depth);
int main()
{
	int n,m;
	cin>>n>>m;//n为节点的总个数，m为 没有叶子结点的个数
	for(int i=0;i<m;i++)
	{
		int id,k,id_;
		cin>>id>>k;
		for(int i=0;i<k;i++)
		{
			cin>>id_;
			tree[id].push_back(id_);
		 } 
	}
	dfs(1,1);
	cout<<num[1];
	for(int i=2;i<=maxdepth;i++)
	{
		cout<<" "<<num[i];
	}

	return 0;
}
void dfs(int node,int depth)
{
	if(tree[node].size()==0)
	{
		num[node]++;
		maxdepth=max(maxdepth,depth);
		return;
	}
	for(int i=0;i<tree[node].size();i++)
	{
		dfs(tree[node][i],depth+1);
	}
}

总结

看到不知道数据的数量而且数据长度不大时，可以尝试利用动态数组vector
dfs深度遍历时，一般是利用递归。

void dfs(int node,int depth)
{
	if(tree[node].size()==0)
	{
		num[node]++;
		maxdepth=max(maxdepth,depth);
		return;
	}
	for(int i=0;i<tree[node].size();i++)
	{
		dfs(tree[node][i],depth+1);
	}
}