記一道簡單的離散化並查集

題目如下

題目背景

現代的人對於本家族血統越來越感興趣。

題目描述

給出充足的父子關係，請你編寫程序找到某個人的最早的祖先。

輸入格式

輸入由多行組成，首先是一系列有關父子關係的描述，其中每一組父子關係中父親只有一行，兒子可能有若干行，用 #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!='$');

}