比较菜的DFS解法。
基本思路就是dfs搜索的时候每进入到子结点时,使深度+1,如果比记录的最大深度大,那么更新一下最大深度。如果存在子结点,那么它就不是叶结点(用flag
标记)。
直接贴代码:
#include <iostream>
using namespace std;
const int MAX = 100;
int G[MAX][MAX] = {0}; /* 邻接矩阵存图,反正数据规模不大 */
int res[MAX] = {0}; /* 最后结果 */
int n, m;
int max_depth = 0; /* 记录最大深度,最后输出需要 */
void dfs(int start, int depth){
int flag = 1; /* 标记是否是叶结点 */
if(depth > max_depth) max_depth = depth; /* 更新最大深度 */
for(int i = 1; i < n+1; ++i){
if(G[start][i] == 1){
flag = 0; /* 存在子节点 */
dfs(i, depth+1); /* 接着搜索子节点 */
}
}
res[depth] += flag; /* 是叶结点就在本层加一 */
}
int main(){
cin >> n >> m;
/* 存图 */
for(int i = 0 ; i < m; ++i){
int num, father;
cin >> father >> num;
for(int j = 0; j < num; ++j){
int son;
cin >> son;
G[father][son] = 1;
}
}
/* 搜索 */
dfs(1, 1);
/* 输出 */
for(int i = 1; i <= max_depth; ++i){
cout << res[i];
if(i != max_depth) cout << ' ';
}
return 0;
}