hihoCoder#1062 最近公共祖先·

原题地址

A和A的共同祖先是A，即使A没有在之前的家谱中出现过！被这个坑了，WA了很久。。。

比如：小头爸爸是大头儿子他爹，问：隔壁王叔叔和隔壁王叔叔的最近祖先是谁？，答：隔壁王叔叔。

代码：

 1 #include <iostream>
 2 #include <map>
 3 #include <set>
 4 #include <string>
 5
 6 using namespace std;
 7
 8 int main() {
 9   int N, M;
10   map<string, string> f;
11
12   cin >> N;
13   for (int i = 1; i <= N; i++) {
14     string father, child;
15     cin >> father >> child;
16     f[child] = father;
17   }
18   cin >> M;
19   while (M--) {
20     string a, b;
21     string ancestor = "-1";
22     set<string> visited;
23     cin >> a >> b;
24     while (!a.empty()) {
25       visited.insert(a);
26       a = f[a];
27     }
28     while (!b.empty()) {
29       if (visited.find(b) != visited.end()) {
30         ancestor = b;
31         break;
32       }
33       b = f[b];
34     }
35     cout << ancestor << endl;
36   }
37
38   return 0;
39 }