动规(23)-并查集基础题——家谱

【问题描述】

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

【输入格式】gen.in

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

【输出格式】gen.out

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

【解题思路】

只是一个普通的并查集问题，然而字符串的处理挺麻烦，然后遇到了神奇的BUG。。。

需要用到映射数组（关联式容器），实现编号到名字、名字到编号的互相转换；

需要注意的是，给名字的编号必须是它第一次出现的编号

样例输入

#George

+Rodney

#Arthur

+Gareth

+Walter

#Gareth

+Edward

?Edward

?Walter

?Rodney

?Arthur

$

样例输出

Edward Arthur

Walter Arthur

Rodney George

Arthur Arthur

#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
using namespace std;
map<string, int> name;
map<int, string> num;
int fa[50005];
string s, ss;
int i, j, k, len, father, t;
int find(int x)
{
    return x == fa[x] ? x : fa[x] = find(fa[x]);
}
int main()
{
    for (i = 1; i <= 50000; ++i)
        fa[i] = i;
    while (cin >> s && s != "$")
    {
        if (s[0] == '#' || s[0] == '+')
        {
            ss = s.substr(1, s.size() - 1);
            if (!name.count(ss))
                name[ss] = ++t, num[t] = ss;
            if (s[0] == '#')
                father = name[ss];
            if (s[0] == '+')
            {
                int f1 = name[ss], f2 = find(father);
                fa[f1] = f2;
            }
        }
        if (s[0] == '?')
        {
            ss = s.substr(1, s.size() - 1);
            cout << ss << " " << num[find(name[ss])] << endl;
        }
    }
    return 0;
}