记一道简单的离散化并查集

题目如下

题目背景

现代的人对于本家族血统越来越感兴趣。

题目描述

给出充足的父子关系，请你编写程序找到某个人的最早的祖先。

输入格式

输入由多行组成，首先是一系列有关父子关系的描述，其中每一组父子关系中父亲只有一行，儿子可能有若干行，用 #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!='$');

}