HDOJ 2586 How far away? LCA

http://acm.hdu.edu.cn/showproblem.php?pid=2586

题意：

给一棵树，多次查询点到点距离。

分析：

d[x][y] = d[x][root] + d[y][root] - 2 * d[lca(x,y)][root]，所以求lca即可。因为tarjan是在dfs的过程中求lca，顺便就把点到根的距离也求了。

敲了代码才发现，讲解的伪代码都是先递归子节点，染色，再回答询问的顺序，但是平时写的时候一般是建了双向边，先递归子节点再染色会导致环的情况从而死循环。所以网上的题解大多是先染色，再查询，再递归子节点。不过我试了一下，至少这一题，先染色，然后先回答询问或是先递归子节点都是可以AC的。

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 #include<vector>
 5 using namespace std;
 6
 7 typedef pair<int, int> P;
 8 const int maxn = 40100;
 9
10 int T, n, m, x, y, z;
11 int f[maxn], d[maxn], ans[300];
12 bool vis[maxn];
13 vector<P> e[maxn], q[maxn];
14
15 int getf(int x)
16 {
17     if (f[x] == x) return x;
18     return f[x] = getf(f[x]);
19 }
20 void unionxy(int x, int y)
21 {
22     int xroot = getf(x), yroot = getf(y);
23     f[yroot] = xroot;
24 }
25 void tarjan(int x)
26 {
27     f[x] = x;
28     vis[x] = true;
29     for (int i = 0; i < q[x].size(); i++){
30         int v = q[x][i].first, id = q[x][i].second;
31         if (vis[v]){
32             int lca = f[getf(v)];
33             ans[id] = d[x] + d[v] - 2 * d[lca];
34         }
35     }
36     for (int i = 0; i < e[x].size(); i++){
37         int v = e[x][i].first, w = e[x][i].second;
38         if (!vis[v]){
39             d[v] = d[x] + w;
40             tarjan(v);
41             unionxy(x, v);
42             f[getf(x)] = x;
43         }
44     }
45 }
46 int main()
47 {
48     scanf("%d", &T);
49     while(T--)
50     {
51         scanf("%d %d", &n, &m);
52         for (int i = 0; i <= n; i++){
53             q[i].clear(); e[i].clear();
54         }
55         for (int i = 1; i < n; i++){
56             scanf("%d%d%d", &x, &y, &z);
57             e[x].push_back(make_pair(y, z));
58             e[y].push_back(make_pair(x, z));
59         }
60         for (int i = 0; i < m; i++){
61             scanf("%d %d", &x, &y);
62             if (x == y) ans[i] = 0;
63             else{
64                 q[x].push_back(make_pair(y, i));
65                 q[y].push_back(make_pair(x, i));
66             }
67         }
68         memset(vis, 0, sizeof(vis));
69         d[1] = 0;
70         tarjan(1);
71         for (int i = 0; i < m; i++)
72             printf("%d\n", ans[i]);
73     }
74     return 0;
75 }

HDOJ 2586 How far away? LCA