hihoCoder#1067 最近公共祖先·二

原题地址

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

需要注意：

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

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

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

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

代码：

#include <iostream>
#include <map>
#include <set>
#include <vector>
#include <cstring>

using namespace std;

#define SIZE 100010

int N, M;
vector<int> tree[SIZE];
int color[SIZE];
int ancestor[SIZE];
vector<pair<int, int> > query[SIZE];
int ans[SIZE];
map<string, int> a2i;
string i2a[SIZE];

int find_gray(int node) {
  if (color[node] == 1)
    return node;
  int gray = find_gray(ancestor[node]);
  ancestor[node] = gray;
  return gray;
}

void traverse(int root) {
  color[root] = 1;
  for (auto q : query[root]) {
    if (ans[q.second] != 0)
      continue;
    if (color[q.first] == 1)
      ans[q.second] = q.first;
    if (color[q.first] == 2)
      ans[q.second] = find_gray(q.first);
  }

  for (auto c : tree[root])
    traverse(c);

  color[root] = 2;
}

int main() {
  memset(color, 0, SIZE * sizeof(int));
  memset(ancestor, 0, SIZE * sizeof(int));
  memset(ans, 0, SIZE * sizeof(int));

  int order = 1;
  cin >> N;
  for (int i = 0; i < N; i++) {
    string f, s;
    int fi, si;
    cin >> f >> s;
    fi = a2i[f];
    si = a2i[s];
    if (fi == 0) {
      fi = a2i[f] = order;
      i2a[order] = f;
      order++;
    }
    if (si == 0) {
      si = a2i[s] = order;
      i2a[order] = s;
      order++;
    }
    tree[fi].push_back(si);
    ancestor[si] = fi;
    color[fi] = color[si] = 0;
  }
  cin >> M;
  for (int i = 0; i < M; i++) {
    string a, b;
    int ai, bi;
    cin >> a >> b;
    ai = a2i[a];
    bi = a2i[b];
    query[ai].push_back(pair<int, int>(bi, i));
    query[bi].push_back(pair<int, int>(ai, i));
  }

  traverse(1);

  for (int i = 0; i < M; i++)
    cout << i2a[ans[i]] << endl;

  return 0;
}