hihoCoder#1067 最近公共祖先·二

原题地址

超时、超内存都碰到了。。最后还是参考了这篇博文才勉强AC

需要注意：

1. 肯定是树而不是森林，而且树的根节点一定是第一个出现的名字，所以不需要再去找哪个是根了。这样可以节省一部分内存。

2. 用并查集路径压缩的方法维护并查集结构即可，当查找的时候再压缩，不需要每次染黑节点的时候都压缩。这样可以节省一部分时间。

3. 字符串都转成id再做。

4. 能静态申请内存就尽量用静态内存，比如数组，STL少用，尤其是map，效率极低。

代码：

 1 #include <iostream>
 2 #include <map>
 3 #include <set>
 4 #include <vector>
 5 #include <cstring>
 6
 7 using namespace std;
 8
 9 #define SIZE 100010
10
11 int N, M;
12 vector<int> tree[SIZE];
13 int color[SIZE];
14 int ancestor[SIZE];
15 vector<pair<int, int> > query[SIZE];
16 int ans[SIZE];
17 map<string, int> a2i;
18 string i2a[SIZE];
19
20 int find_gray(int node) {
21   if (color[node] == 1)
22     return node;
23   int gray = find_gray(ancestor[node]);
24   ancestor[node] = gray;
25   return gray;
26 }
27
28 void traverse(int root) {
29   color[root] = 1;
30   for (auto q : query[root]) {
31     if (ans[q.second] != 0)
32       continue;
33     if (color[q.first] == 1)
34       ans[q.second] = q.first;
35     if (color[q.first] == 2)
36       ans[q.second] = find_gray(q.first);
37   }
38
39   for (auto c : tree[root])
40     traverse(c);
41
42   color[root] = 2;
43 }
44
45 int main() {
46   memset(color, 0, SIZE * sizeof(int));
47   memset(ancestor, 0, SIZE * sizeof(int));
48   memset(ans, 0, SIZE * sizeof(int));
49
50   int order = 1;
51   cin >> N;
52   for (int i = 0; i < N; i++) {
53     string f, s;
54     int fi, si;
55     cin >> f >> s;
56     fi = a2i[f];
57     si = a2i[s];
58     if (fi == 0) {
59       fi = a2i[f] = order;
60       i2a[order] = f;
61       order++;
62     }
63     if (si == 0) {
64       si = a2i[s] = order;
65       i2a[order] = s;
66       order++;
67     }
68     tree[fi].push_back(si);
69     ancestor[si] = fi;
70     color[fi] = color[si] = 0;
71   }
72   cin >> M;
73   for (int i = 0; i < M; i++) {
74     string a, b;
75     int ai, bi;
76     cin >> a >> b;
77     ai = a2i[a];
78     bi = a2i[b];
79     query[ai].push_back(pair<int, int>(bi, i));
80     query[bi].push_back(pair<int, int>(ai, i));
81   }
82
83   traverse(1);
84
85   for (int i = 0; i < M; i++)
86     cout << i2a[ans[i]] << endl;
87
88   return 0;
89 }