[PAT解题报告] The Largest Generation
简单题,给定一棵树——树是按照父子关系给出的,求节点个数最多的那层有多少个节点——保证这层是唯一的。并且认为根的层数为1。
因为数非空,并且规定了根是1号,我把id都减去1之后,就认为第0层是根。用d[]表示每个节点的层数,即深度。用num[]表示每层的节点数。别忘记初值d[0]
= 1, num[1] = 1。
读入树结构的时候,我们记录每个节点x的父亲y,即f[x] = y。 则d[x] = d[y] +
1。我们dfs求出每个节点的深度——注意如果之前求过,递归可以直接返回,所以计算深度是O(n)的,相当于dfs整个树。我用了全局变量,没求过的深度暂且认为0。当然也可以从根直接dfs一次直接得到每个节点的深度,写起来比我现在这个稍微麻烦一点,总之可以求出每个节点的深度,顺便累加到num[d[x]]上面去,就能求出节点数最多的层了。
代码;
#include <cstdio>
#include <cstring>
#include <string>
using namespace std;
int f[105];
int d[105] = {1};
int num[105] = {0, 1};
int getd(int i) {
return d[i]?d[i]:(d[i] = getd(f[i]) + 1);
}
int main() {
int n, m;
for (scanf("%d%d", &n,&m);m;--m) {
int x, i;
scanf("%d%d",&x,&i);
for (--x; i; --i) {
int y;
scanf("%d", &y);
f[y - 1] = x;
}
}
m = 1;
for (int i = 1; i < n; ++i) {
int temp = getd(i);
if (++num[temp] > num[m]) {
m = temp;
}
}
printf("%d %d\n", num[m], m);
return 0;
}
原题链接: http://www.patest.cn/contests/pat-a-practise/1094
神州信息成长空间 29人发布