算法题：家谱

问题描述

现代的人对于本家族血统越来越感兴趣，现在给出充足的父子关系，请你编写程序找到某个人的最早的祖先。

输入格式

输入文件由多行组成，首先是一系列有关父子关系的描述，其中每一组父子关系由二行组成，用#name的形式描写一组父子关系中的父亲的名字，用+name的形式描写一组父子关系中的儿子的名字；接下来用?name的形式表示要求该人的最早的祖先；最后用单独的一个$表示文件结束。规定每个人的名字都有且只有6个字符，而且首字母大写，且没有任意两个人的名字相同。最多可能有1000组父子关系，总人数最多可能达到50000人，家谱中的记载不超过30代。

输出格式

按照输入文件的要求顺序，求出每一个要找祖先的人的祖先，格式：本人的名字+一个空格+祖先的名字+回车。

输入

#George
+Rodney
#Arthur
+Gareth
+Walter
#Gareth
+Edward
?Edward
?Walter
?Rodney
?Arthur
$

输出

Edward Arthur
Walter Arthur
Rodney George
Arthur Arthur

解决方案

思路

并查集

代码

#include <iostream>
#include <map>

using namespace std;

map<string, string> fa;

int main() {
    char c;
    string faName;
    string name;
    while (true) {
        cin >> c;
        if (c == '$')
            return 0;

        cin >> name;
        switch (c) {
            case '#':
                if (fa[name] == "")
                    fa[name] = name;
                faName = name;
                break;
            case '+':
                fa[name] = faName;
                break;
            case '?':
                string buf = name;
                while (fa[buf] != buf) {
                    buf = fa[buf];
                }

                cout << name << " " << buf << endl;
                break;
        }
    }

    return 0;
}