HDU 2586 How far away?

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

题意：给定一个图，有M个询问。每一次询问为询问u，v两个点的距离为多少？

题解：LCA问题。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <string>
#include <vector>
#include <list>
#include <map>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <numeric>
#include <functional>
#include <set>
#include <fstream>

using namespace std;

const int maxn=40010;

struct edge{
    int from,to,cost,next;
};
edge es[maxn*2];
int head[maxn],idx;
int dis[maxn],N,M;
int U[maxn],V[maxn];
int par[maxn],rankh[maxn];
int LCA[maxn];
bool used[maxn];

void add_edge(int u,int v,int cost)
{
    es[idx].from=u;
    es[idx].to=v;
    es[idx].cost=cost;
    es[idx].next=head[u];
    head[u]=idx++;
}

void build_graph(int u,int v,int cost)
{
    add_edge(u,v,cost);
    add_edge(v,u,cost);
}

void init(int n)
{
    for(int i=0;i<=n;i++)
    {
        par[i]=i;
        rankh[i]=0;
    }
}

int find(int x)
{
    if(x==par[x]) return x;
    else return par[x]=find(par[x]);
}

void tarjan(int u)
{
    used[u]=true;
    for(int  i=0;i<M;i++)
    {
        if(U[i]==u&&used[V[i]]) LCA[i]=find(V[i]);
        if(V[i]==u&&used[U[i]]) LCA[i]=find(U[i]);
    }
    for(int i=head[u];i!=-1;i=es[i].next)
    {
        int v=es[i].to;
        if(!used[v]){
            dis[v]=dis[u]+es[i].cost;
            tarjan(v);
            par[v]=u;
        }
    }
}

int main()
{
//    freopen("/Users/apple/Desktop/暑假/HDU 2586/HDU 2586/in","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&N,&M);
        init(2*N);
        memset(head,-1,sizeof(head));
        idx=0;
        memset(used,false,sizeof(used));
        dis[1]=0;
        for(int i=0;i<=N;i++)
        {
            U[i]=V[i]=LCA[i]=0;
        }
        for(int i=0;i<N-1;i++)
        {
            int u,v,cost;
            scanf("%d%d%d",&u,&v,&cost);
            build_graph(u, v, cost);
        }
        for(int i=0;i<M;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            U[i]=u,V[i]=v;
        }
        tarjan(1);
        for(int i=0;i<M;i++)
        {
            int res=dis[U[i]]+dis[V[i]]-2*dis[LCA[i]];
            printf("%d
",res);
        }
    }
    return 0;
}