題目如下
題目背景
現代的人對於本家族血統越來越感興趣。
題目描述
給出充足的父子關係,請你編寫程序找到某個人的最早的祖先。
輸入格式
輸入由多行組成,首先是一系列有關父子關係的描述,其中每一組父子關係中父親只有一行,兒子可能有若干行,用 #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!='$');
}