L2-026. 小字辈
时间限制
400 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越
本题给定一个庞大家族的家谱,要请你给出最小一辈的名单。
输入格式:
输入在第一行给出家族人口总数 N(不超过 100 000 的正整数) —— 简单起见,我们把家族成员从 1 到 N 编号。随后第二行给出 N 个编号,其中第 i 个编号对应第 i 位成员的父/母。家谱中辈分最高的老祖宗对应的父/母编号为 -1。一行中的数字间以空格分隔。
输出格式:
首先输出最小的辈分(老祖宗的辈分为 1,以下逐级递增)。然后在第二行按递增顺序输出辈分最小的成员的编号。编号间以一个空格分隔,行首尾不得有多余空格。
输入样例:9 2 6 5 5 -1 5 6 4 7输出样例:
4 1 9
考试当天也就一遍AC了,后来重写了一遍居然花了一个多小时(考场效率,果然deadline是第一生产线
这题我用的方法是建立一个“树状关系”
建立起父子关系 子们用vector去存储 读取的时候使用C风格的下标读取
通过结构体数组等各种关系去链接 建立关系
#include<iostream>
#include<vector>
using namespace std;
const int maxn=100000+5;
int arr[maxn]={0};
int N;
struct rela{
int level;
int fa;
vector<int>child;
};
//分别用于存储
rela pel[maxn];
int mmaxn=1;
void init(){
for(int i=0;i<=N;i++){
pel[i].level=0;
pel[i].fa=0;
}
}
void bfs(int mem){
for(vector<int>::size_type i=0;i<pel[mem].child.size();i++){
pel[pel[mem].child[i]].level=pel[pel[pel[mem].child[i]].fa].level+1;//mem所对应的每一个孩子的等级=孩子的爸爸的等级+1
if(pel[pel[mem].child[i]].level>mmaxn)//记录下最大的序号
mmaxn=pel[pel[mem].child[i]].level;
}
for(vector<int>::size_type i=0;i<pel[mem].child.size();i++)
bfs(pel[mem].child[i]);
}
int main(){
cin>>N;
for(int i=1;i<=N;i++)
cin>>arr[i];
int mem=0;
for(int i=1;i<=N;i++){
pel[i].fa=arr[i];
if(arr[i]==-1){
pel[i].level=1;
mem=i;
}
else{
pel[arr[i]].child.push_back(i);
}
}
bfs(mem);
//for(int i=1;i<=N;i++){
// cout<<i<<"'s father is "<<pel[i].fa<<" level: "<<pel[i].level<<endl;
//}
int count=0;
cout<<mmaxn<<endl;
for(int i=1;i<=N;i++){
if(pel[i].level==mmaxn){
if(count==0){
cout<<i;
count++;
}
else
cout<<" "<<i;
}
}
return 0;
}
然而这个算法并没有超时,可能是“空间换时间”,所以前n次都挂了,空间太大。
之前的写法和比赛的时候一样 也是用的数组和显性的长度
挂掉的代码如下:
#include<iostream>
using namespace std;
const int maxn=20000;
int arr[maxn]={0};
int N;
struct rela{
int level;
int fa;
int child[maxn];
int k;
};
rela pel[maxn];
int mmaxn=0;
void init(){
for(int i=1;i<=N;i++){
pel[i].level=0;
pel[i].k=0;
pel[i].fa=0;
}
}
void bfs(int mem){
for(int i=0;i<pel[mem].k;i++){
pel[pel[mem].child[i]].level=pel[pel[pel[mem].child[i]].fa].level+1;
if(pel[pel[mem].child[i]].level>mmaxn)
mmaxn=pel[pel[mem].child[i]].level;
}
for(int i=0;i<pel[mem].k;i++)
bfs(pel[mem].child[i]);
}
int main(){
cin>>N;
for(int i=1;i<=N;i++)
cin>>arr[i];
int mem=0;
for(int i=1;i<=N;i++){
pel[i].fa=arr[i];
if(arr[i]==-1){
pel[i].level=0;
mem=i;
}
else{
int temp;
temp=pel[arr[i]].k;
pel[arr[i]].k+=1;
pel[arr[i]].child[temp]=i;
}
}
pel[mem].level=1;
bfs(mem);
//for(int i=1;i<=N;i++){
// cout<<i<<"'s father is "<<pel[i].fa<<" level: "<<pel[i].level<<endl;
//}
int count=0;
cout<<mmaxn<<endl;
for(int i=1;i<=N;i++){
if(pel[i].level==mmaxn)
if(count==0){
cout<<i;
count++;
}
else
cout<<" "<<i;
}
return 0;
}
尽量避免这种写法,使用STL最为简洁和安全。