L2-026. 小字辈(vector 模拟)

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最为简洁和安全。