[BZOJ1602] [Usaco2008 Oct] 牧场行走 (LCA)

Description

　　N头牛（2<=n<=1000）别人被标记为1到n，在同样被标记1到n的n块土地上吃草，第i头牛在第i块牧场吃草。 这n块土地被n-1条边连接。 奶牛可以在边上行走，第i条边连接第Ai，Bi块牧场，第i条边的长度是Li（1<=Li<=10000）。 这些边被安排成任意两头奶牛都可以通过这些边到达的情况，所以说这是一棵树。 这些奶牛是非常喜欢交际的，经常会去互相访问,他们想让你去帮助他们计算Q(1<=q<=1000)对奶牛之间的距离。

Input

　　*第一行：两个被空格隔开的整数：N和Q

　　*第二行到第n行：第i+1行有两个被空格隔开的整数：AI，BI，LI

　　*第n+1行到n+Q行：每一行有两个空格隔开的整数：P1，P2，表示两头奶牛的编号。

Output

　　*第1行到第Q行：每行输出一个数，表示那两头奶牛之间的距离。

Sample Input

4 2
2 1 2
4 3 2
1 4 3
1 2
3 2

Sample Output

2
7

HINT

Source

　　资格赛

Solution

　　裸LCA，看不懂代码的人都是　　　　

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 struct edge
 4 {
 5     int v, w, nxt;
 6 }e[2005];
 7 struct query
 8 {
 9     int u, v, nxt;
10 }q[2005];
11 int efst[1005], qfst[1005], fa[1005], lca[1005], dis[1005];
12 bool vis[1005];
13
14 void addedge(int i, int u, int v, int w)
15 {
16     e[i] = (edge){v, w, efst[u]}, efst[u] = i;
17 }
18
19 void addquery(int i, int u, int v)
20 {
21     q[i] = (query){u, v, qfst[u]}, qfst[u] = i;
22 }
23
24 int get_dis(int i)
25 {
26     return dis[q[i << 1].u] + dis[q[i << 1].v] - 2 * dis[lca[i]];
27 }
28
29 int getfa(int x)
30 {
31     return fa[x] = x == fa[x] ? x : getfa(fa[x]);
32 }
33
34 void Tarjan(int u)
35 {
36     fa[u] = u, vis[u] = true;
37     for(int i = efst[u]; i; i = e[i].nxt)
38         if(!vis[e[i].v])
39         {
40             dis[e[i].v] = dis[u] + e[i].w;
41             Tarjan(e[i].v);
42             fa[e[i].v] = u;
43         }
44     for(int i = qfst[u]; i; i = q[i].nxt)
45     {
46         int v = q[i].u == u ? q[i].v : q[i].u;
47         if(vis[v]) lca[i >> 1] = getfa(fa[v]);
48     }
49 }
50
51 int main()
52 {
53     int n, q, u, v, w;
54     cin >> n >> q;
55     for(int i = 1; i < n; i++)
56     {
57         cin >> u >> v >> w;
58         addedge(i << 1, u, v, w);
59         addedge(i << 1 | 1, v, u, w);
60     }
61     for(int i = 1; i <= q; i++)
62     {
63         cin >> u >> v;
64         addquery(i << 1, u, v);
65         addquery(i << 1 | 1, v, u);
66     }
67     Tarjan(1);
68     for(int i = 1; i <= q; i++)
69         cout << get_dis(i) << endl;
70     return 0;
71 }