PTA | 小字辈

目录

题目：

输入格式：

输出格式：

输入样例：

输出样例：

代码：

无注释版：

有注释版： 

---

题目：

本题给定一个庞大家族的家谱，要请你给出最小一辈的名单。

输入格式：

输入在第一行给出家族人口总数 N（不超过 100 000 的正整数） —— 简单起见，我们把家族成员从 1 到 N 编号。随后第二行给出 N 个编号，其中第 i 个编号对应第 i 位成员的父/母。家谱中辈分最高的老祖宗对应的父/母编号为 -1。一行中的数字间以空格分隔。

输出格式：

首先输出最小的辈分（老祖宗的辈分为 1，以下逐级递增）。然后在第二行按递增顺序输出辈分最小的成员的编号。编号间以一个空格分隔，行首尾不得有多余空格。

输入样例：

9
2 6 5 5 -1 5 6 4 7

输出样例：

4
1 9

代码长度限制16 KB，时间限制400 ms，内存限制64 MB，栈限制8192 KB

代码：

无注释版：

#include<bits/stdc++.h>
using namespace std;
int f[100010];
int a[100010];
int b[100010];
int n;
int find(int x){
	if(f[x]==-1)  return a[x]=1;
	if(a[x]==0) return a[x]=find(f[x])+1;
	return a[x];
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>f[i];
	}
	int ma=0;
	for(int i=1;i<=n;i++){
		ma=max(ma,find(i));
	}
	int m=0;
	for(int i=1;i<=n;i++){
		if(a[i]==ma){
			b[++m]=i;
		}
	}
	sort(b+1,b+m+1);
	cout<<ma<<"\n";
	for(int i=1;i<=m;i++){
		cout<<b[i];
		if(i!=m) cout<<" ";
	}
} 

有注释版： 

#include<bits/stdc++.h>  // 引入所有标准库，简化代码书写
using namespace std;     // 使用标准命名空间，避免使用 std:: 前缀

int f[100010];  // f 数组用于存储每个成员的父/母编号
int a[100010];  // a 数组用于存储每个成员的辈分信息
int b[100010];  // b 数组用于存储最小辈分成员的编号
int n;  // n 表示家族总人口数

// 递归查找成员 x 的辈分信息
int find(int x) {
    if (f[x] == -1)  // 如果成员 x 的父母编号是 -1，说明是老祖宗，辈分为 1
        return a[x] = 1;  // 直接返回辈分 1
    if (a[x] == 0)  // 如果成员 x 还没有计算过辈分
        return a[x] = find(f[x]) + 1;  // 递归查找父母的辈分，并加 1 得到当前成员的辈分
    return a[x];  // 如果成员 x 已经计算过辈分，直接返回该辈分
}

int main() {
    cin >> n;  // 输入家族人口总数 n

    // 输入每个成员的父母编号，存储在 f 数组中
    for (int i = 1; i <= n; i++) {
        cin >> f[i];  // f[i] 为第 i 位成员的父母编号，-1 表示是老祖宗
    }

    int ma = 0;  // ma 变量用于存储最大的辈分
    // 计算所有成员的辈分
    for (int i = 1; i <= n; i++) {
        ma = max(ma, find(i));  // 通过 find 函数更新最大辈分
    }

    int m = 0;  // m 变量用于记录最小辈分成员的个数
    // 找出所有辈分等于 ma 的成员，并存储在 b 数组中
    for (int i = 1; i <= n; i++) {
        if (a[i] == ma) {  // 如果成员 i 的辈分等于最大辈分
            b[++m] = i;  // 将成员 i 编号加入 b 数组
        }
    }

    sort(b + 1, b + m + 1);  // 按照成员编号升序排列 b 数组

    cout << ma << "\n";  // 输出最大辈分
    for (int i = 1; i <= m; i++) {  // 输出所有最大辈分成员的编号
        cout << b[i];  // 输出成员编号
        if (i != m) cout << " ";  // 如果不是最后一个成员，则输出空格
    }
}