本题给定一个庞大家族的家谱，要请你给出最小一辈的名单。
输入格式：
输入在第一行给出家族人口总数 N（不超过 100 000 的正整数） —— 简单起见，我们把家族成员从 1 到 N 编号。随后第二行给出 N 个编号，其中第 i 个编号对应第 i 位成员的父/母。家谱中辈分最高的老祖宗对应的父/母编号为 -1。一行中的数字间以空格分隔。
输出格式：
首先输出最小的辈分（老祖宗的辈分为 1，以下逐级递增）。然后在第二行按递增顺序输出辈分最小的成员的编号。编号间以一个空格分隔，行首尾不得有多余空格。
输入样例：
9
2 6 5 5 -1 5 6 4 7
输出样例：
4
1 9
原始做法是以叶子节点往上查 由于时限为400ms所以不可取
#include<bits/stdc++.h>
using namespace std;
int s[100000 + 10], p[100000 + 10];
int main() {
//freopen("in.txt", "r", stdin);
int n;
memset(p, 0, sizeof(p));
scanf("%d", &n);
for (int i = 1; i <= n; i++) {
scanf("%d", &s[i]);
if (s[i] != -1)
p[s[i]] = 1;
}
int ad = 1, ans[100000 + 10],m=0;
for (int i = 1; i <= n; i++) {
if (!p[i]) {
int t = 1, a = i;
while (s[a] != -1) {
t++;
a = s[a];
}
if (ad < t) {
ad = t;
m = 0;
ans[m++] = i;
}
else if (ad == t)
ans[m++] = i;
}
}
if (ad == 1) printf("1\n1\n");
else {
printf("%d\n%d", ad,ans[0]);
for (int i = 1; i < m; i++)
printf(" %d", ans[i]);
}
return 0;
}
这是用根的方法
#include<bits/stdc++.h>
using namespace std;
#define maxn 100000 + 10
vector<int>s[maxn];
int ans[maxn], f, ad = 1, m = 0;
void dfs(int x, int d) {
int n = s[x].size();
if (n == 0) {
if (ad < d) {
ad = d;
m = 0;
ans[m++] = x;
}
else if (ad == d)
ans[m++] = x; }
else
for (int i = 0, t; i < n; i++) {
dfs(s[x][i], d + 1);
} }
int main() {
//freopen("in.txt", "r", stdin);
int n;
scanf("%d", &n);
for (int i = 1, t; i <= n; i++) {
scanf("%d", &t);
if (t == -1)
f = i;
else
s[t].push_back(i);
}
dfs(f, 1);
if (ad == 1) printf("1\n1\n");
else {
printf("%d\n%d", ad, ans[0]);
for (int i = 1; i < m; i++)
printf(" %d", ans[i]);
}
return 0;
}