题意:求一颗树节点个数最多的层次以及个数
思路:深度搜索树,result是存放每层节点个数的数组,在深度搜索的时候就完成对result数组的赋值
注意:可能只有一个节点,故ans至少为1,只有一个节点ansid为1,1为根节点
代码:
#include<iostream>
#include<vector>
using namespace std;
vector<vector<int>> v;
int result[100];
void DFS(int root,int level) {
result[level]++;
for (int i = 0; i < v[root].size(); i++)
DFS(v[root][i],level+1);
}
int main() {
int n = 0, m = 0,root = 0,cnt = 0,tem = 0,ans = 1,ansid = 1;
cin >> n >> m;
v.resize(n+1);
for (int i = 0; i < m; i++) {
scanf("%d%d",&root,&cnt);
for (int j = 0; j < cnt; j++) {
scanf("%d",&tem);
v[root].push_back(tem);
}
}
DFS(1,1);
ans = result[1];
for (int i = 1; i <= n; i++) {
if (result[i] > ans) {
ans = result[i];
ansid = i;
}
}
printf("%d %d\n",ans,ansid);
system("pause");
return 0;
}