[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，看不懂代码的人都是　　　　

#include <bits/stdc++.h>
using namespace std;
struct edge
{
    int v, w, nxt;
}e[2005];
struct query
{
    int u, v, nxt;
}q[2005];
int efst[1005], qfst[1005], fa[1005], lca[1005], dis[1005];
bool vis[1005];
 
void addedge(int i, int u, int v, int w)
{
    e[i] = (edge){v, w, efst[u]}, efst[u] = i;
}
 
void addquery(int i, int u, int v)
{
    q[i] = (query){u, v, qfst[u]}, qfst[u] = i;
}
 
int get_dis(int i)
{
    return dis[q[i << 1].u] + dis[q[i << 1].v] - 2 * dis[lca[i]];
}
 
int getfa(int x)
{
    return fa[x] = x == fa[x] ? x : getfa(fa[x]);
}
 
void Tarjan(int u)
{
    fa[u] = u, vis[u] = true;
    for(int i = efst[u]; i; i = e[i].nxt)
        if(!vis[e[i].v])
        {
            dis[e[i].v] = dis[u] + e[i].w;
            Tarjan(e[i].v);
            fa[e[i].v] = u;
        }
    for(int i = qfst[u]; i; i = q[i].nxt)
    {
        int v = q[i].u == u ? q[i].v : q[i].u;
        if(vis[v]) lca[i >> 1] = getfa(fa[v]);
    }
}
 
int main()
{
    int n, q, u, v, w;
    cin >> n >> q;
    for(int i = 1; i < n; i++)
    {
        cin >> u >> v >> w;
        addedge(i << 1, u, v, w);
        addedge(i << 1 | 1, v, u, w);
    }
    for(int i = 1; i <= q; i++)
    {
        cin >> u >> v;
        addquery(i << 1, u, v);
        addquery(i << 1 | 1, v, u);
    }
    Tarjan(1);
    for(int i = 1; i <= q; i++)
        cout << get_dis(i) << endl;
    return 0;
}