1388：家谱(gen)(并查集+字符串处理)

1388：家谱(gen)

时间限制: 1000 ms         内存限制: 65536 KB
提交数: 4704     通过数: 2415

【题目描述】

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

【输入】

由多行组成，首先是一系列有关父子关系的描述，其中每一组父子关系由二行组成，用#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

---

思路：

直接无脑跟题目模拟，用一个vec数组存储曾经出现过的人,输入一个人名s后先用一个临时string tk存储s,再删掉第一个操作字符(只保留剩下的人名)后在vec中查找tk以前有没有出现,如果没有则将fa[tk] = tk(fa是一个map),如果有则不需要做上述操作,然后判断s[0]是否为'?',是则将s的第一位删除再输出s和fin(s)，如果s[0]=='#'则将将s的第一位删除再将f = s,否则说明s[0]为‘+'，则删除s[0],再join(s,f)即可。

---

#include <bits/stdc++.h>
using namespace std;
map<string,string>fa;
string s,f,tk;
char c;
vector<string>vec;
string fin(string k)
{
  if(k == fa[k]) return k;
  return fin(fa[k]);
}
void join(string p,string q)
{
  string fp = fin(p),fq = fin(q);
  if(fp != fq) fa[fp] = fq;
}
int main()
{
  while(cin>>s)
  {
    tk = s;
    tk.erase(0,1);
    bool kj = 1;
    for(int i = 0; i < vec.size(); i++)
      if(vec[i] == tk)
        kj = 0;
    if(kj == 1)
    {
      fa[tk] = tk;
      vec.push_back(tk);
    }
    if(s[0] == '$') break;
    if(s[0] == '?')
    {
      s.erase(0,1);
      cout<<s<<" "<<fin(s)<<endl;
    }
    else if(s[0] == '#')
    {
      s.erase(0,1);
      f = s;
    }
    else if(s[0] == '+')
    {
      s.erase(0,1);
      join(s,f);
    }
    //for(auto i:fa) cout<<i.first<<"_"<<i.second<<" ";
    //cout<<endl;
  }
  return 0;
}