Input Specification:
Each input file contains one test case. Each case starts with a line containing 0<N<100, the number of nodes in a tree, and M (<N), the number of non-leaf nodes. Then M lines follow, each in the format:
ID K ID[1] ID[2] ... ID[K]
where ID is a two-digit number representing a given non-leaf node, K is the number of its children, followed by a sequence of two-digit ID's of its children. For the sake of simplicity, let us fix the root ID to be 01.
The input ends with N being 0. That case must NOT be processed.
Output Specification:
For each test case, you are supposed to count those family members who have no child for every seniority level starting from the root. The numbers must be printed in a line, separated by a space, and there must be no extra space at the end of each line.
The sample case represents a tree with only 2 nodes, where 01 is the root and 02 is its only child. Hence on the root 01 level, there is 0 leaf node; and on the next level, there is 1 leaf node. Then we should output 0 1 in a line.
Sample Input:
2 1
01 1 02
Sample Output:
0 1
题目的大致意思是:首先输出一行,两个数:第一个数是树的所有结点的个数(N),第二个数是非叶子结点的个数(M)。下面的M行每一行输入的是非叶子结点的序号,该非叶子结点的孩子的个数,孩子的序号。(注意规定根节点的序号为01)。要求输出每一层的叶子结点的个数
采用深度优先搜索
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int book[100]={0},maxdepth=-1;//book数组用来标记每一层叶子节点的个数。maxdepth表示该树的最大层数
vector<int> v[100];//创建一个动态数组用于存放非叶子节点的孩子
void dfs(int index,int depth){//深度优先搜索。传入的参数为非叶子结点的序号,和当前的层数
if(v[index].size()==0){//判断当前是否为叶子结点(v[index].size()==0表示没有孩子)
book[depth]++;//对应层叶子节点的个数++
maxdepth=max(maxdepth,depth);//更新最大层次
return;//返回上一次调用函数的地方
}
for(int i=0;i<v[index].size();i++){
dfs(v[index][i],depth+1);//v[index][i]表示的孩子的序号(用二维数组来存入孩子的序号)
}
}
int main()
{
int N,M,k,c,node;//N表示树结点的个数,M表示有几个非叶子结点,k表示孩子的个数,c表示的孩子的序号,node表示非叶子结点的序号
cin>>N>>M;
for(int i=0;i<M;i++){
cin>>node>>k;
for(int j=0;j<k;j++){//输入k个孩子的序号
cin>>c;
v[node].push_back(c);//下标表示非叶子结点的序号,将孩子存入数组中
}
}
dfs(01,0);//深度优先搜索
cout<<book[0];
for(int i=1;i<=maxdepth;i++)
cout<<" "<<book[i];
return 0;
}
- 思路:首先实现输入,要考虑的是如何将非叶子结点和孩子关联起来:第一种是定义结构体,第二种是采用数组下标和数组元素的方式。这里采用第二种方式,将数组的索引定义为非叶子结点的序号,数组元素为孩子结点的序号,定义一个二维数组。然后实现输入。然后考虑如何进行查找判断每一层的非叶子结点的个数。即采用深度优先搜索的方法:首先确定变量是什么:即非叶子结点的序号和层数,所以dfs的形参就是非叶子结点的序号和层数。考虑如何构建dfs。
- dfs的基本模型如下
void dfs(int step)
{
判断边界
尝试每种可能 for(i=1;i<=n;i++)
{
继续下一步dfs(step+1);
}
返回
}
- 所以首先明确该函数的目的是什么:即找到每一层的叶子结点的个数。再考虑边界是什么。边界就是查找到叶子结点,叶子结点的判断标准就是没有孩子,那么其数组长度就为0->
v[index].size()==0
.尝试每一种可能:就是对每个结点进行遍历。就从最开始的根节点开始遍历其孩子(因为每判断依次孩子就会重新调用依次函数进行对当前结点的判断,所以一定可以将所有结点遍历判断完)。最后当遍历完根节点的孩子之后退出函数,完成dfs。 - push_back()函数:就是将一个新的元素加入到vector的最后面,位置为当前最后一个元素的下一个元素。也是用vector数组创建二维数组的方法:
vector<int>a;
a.push_back(1);
a.push_back(2);
a.push_back(3);
- 根据本题对dfs(深度优先搜索)的理解:
1.深度优先搜索的模型:
void dfs(int step)
{
判断边界
尝试每种可能 for(i=1;i<=n;i++)
{
继续下一步dfs(step+1);
}
返回
}
2.构建思路:在构建dfs之前先明确目的是什么,然后判断边界条件(根据目的来判断,也是返回上一层函数的条件,就如本题:对于每个结点进行遍历,当遍历到叶子结点的时候就没有办法再向下进行搜索了,所以这个时候就要向上走一步步返回,看另外一条路,这个就是本题的边界)。尝试每种可能,然后对每一个可能再进行搜索判断,所以在遍历到每一种可能的时候都要再调用函数dfs
3.根据树进行理解:dfs就是一种“不撞南墙不回头”的搜索方法。就是一条路走到底直到到头了(撞南墙)才再一步步退回最顶层,在退回顶层的时候,对于每一步又会再查看是否有别的分支可以走,如果有,就继续按照那个分支走到头(撞南墙),然后继续一步步退回,并进行判断有无其他分支可走…这样进行循环(递归)。直到尝试完所有的分支(道路)。就很像一颗树,首先选择一个分支,一直走到最底层(即走到叶子节点处),然后再看其他的分支,直到把树中的分支全部走完,然后退出。
4.注意要点:创建dfs时明确目的,并把要实现的目的在判断边界的部分当中实现。在主函数中调用dfs时,传入的参数为最开始进行判断的一个具体值(因为在dfs中会从第一个数进行遍历,遍历完所有可能,所以只需要让dfs直到第一个要尽心搜索的是谁即可)。