题目如下
题目背景
现代的人对于本家族血统越来越感兴趣。
题目描述
给出充足的父子关系,请你编写程序找到某个人的最早的祖先。
输入格式
输入由多行组成,首先是一系列有关父子关系的描述,其中每一组父子关系中父亲只有一行,儿子可能有若干行,用 #name
的形式描写一组父子关系中的父亲的名字,用 +name
的形式描写一组父子关系中的儿子的名字;接下来用 ?name
的形式表示要求该人的最早的祖先;最后用单独的一个 $
表示文件结束。
输出格式
按照输入文件的要求顺序,求出每一个要找祖先的人的祖先,格式为:本人的名字 +一个空格 + 祖先的名字 + 回车。
输入输出样例
输入 #1
#George +Rodney #Arthur +Gareth +Walter #Gareth +Edward ?Edward ?Walter ?Rodney ?Arthur $
输出 #1
Edward Arthur Walter Arthur Rodney George Arthur Arthur
说明/提示
规定每个人的名字都有且只有 6个字符,而且首字母大写,且没有任意两个人的名字相同。最多可能有 1e3组父子关系,总人数最多可能达到 5e4 人,家谱e中的记载不超过 30 代。
AC代码如下
#include<bits/stdc++.h>
using namespace std;
const int MAX = 5e5+55;
int father[MAX+5];
char name[MAX][10];
int countName=0;
int getFather(int x) {
if(x==father[x]) return x;
else return father[x]=getFather(father[x]);
}
int getIndex(char* na){
for (int i = 0; i < countName; ++i) {
if(strcmp(na,name[i])==0) return i;
}
return countName;
}
int main() {
char f;
for (int i = 0; i <= MAX; ++i) {
father[i]=i;
}
int fa = 0;
while (cin>>f,f!='?') {
char t[10];
cin >> t;
int index = getIndex(t);
if(index==countName) {
for (int i = 0; i < 6; ++i) {
name[countName][i]=t[i];
}
name[countName][6]='\0';
countName++;
}
if(f=='#') {
fa=index;
} else {
father[index]=fa;
}
}
do {
char t[10];
cin >> t;
cout << t << " ";
int index = getIndex(t);
cout << name[getFather(index)] << endl;
} while (cin >> f,f!='$');
}