Tree(LCA模板)

题目:
给出一棵带有边权的树，问两点之间的距离。

输入:
第一行两个整数 n 和 m ，分别表示点数和询问数。
接下来 n-1 行，每行三个整数 x，y，z，表示 x 与 y 通过一条权为 z 的边连接。
接下来 m 行，每行两个整数 x，y，代表一组询问。

输出:
输出 m 行，每行一个整数，对应一组询问的答案。

EG:
in:
3 3
1 2 1
1 3 2
1 2
1 3
2 3

out:
1
2
3

备注:
【数据范围】
对 30% 的输入数据 ：1≤n,m≤1000
对 100% 的输入数据 ：1≤n,m≤100000；1≤z≤10000

代码:

#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<cstdlib>
#include<ctime>
#include<cctype>
#include<queue>
#include<iomanip>
#include<set>
#include<map>
#include<cmath>
#include<algorithm>
using namespace std;

struct node 
{
     node *next;
     int place,cost;
} *first[100005],a[200005];

int n,m;
int f[100005][25],dist[100005],D[100005],c[100005];
int l;

inline void add(int x,int y,int z) 
{
    ++l;
    a[l].place=y; 
    a[l].cost=z;
    a[l].next=first[x]; 
    first[x]=&a[l];
}

int LCA(int x, int y) 
{
    if(D[x]<D[y]) 
        swap(x,y);

    int will=D[x]-D[y];

    for(int step=0;will;will>>=1,++step)
        if(will&1)
            x=f[x][step];

    if(x==y)
        return x;

    for(int i=20;i>=0;--i)
        if(f[x][i]!=f[y][i]) 
        {   
            x=f[x][i];
            y=f[y][i];
        }

    return f[x][0];
}

int main() 
{
    memset(f, 0, sizeof(f));
    memset(dist, 255, sizeof(dist));
    memset(first, 0, sizeof(first));

    cin>>n>>m;
    l=0;

    for (int i=1;i<n;++i) 
    {
        int x,y,z;

        cin>>x>>y>>z;

        add(x,y,z);
        add(y,x,z);
    }

    dist[1]=0; 
    c[1]=1; 
    D[1]=0;

    for(int k=1,l=1;l<=k;++l) 
    {
        int m=c[l];

        for(node *x=first[m];x;x=x->next)
            if(dist[x->place]==-1) 
            {
                D[x->place]=D[m]+1;
                dist[x->place]=dist[m]+x->cost;
                f[x->place][0]=m;
                c[++k]=x->place;
            }
     }

    for(int i=1;i<=20;++i) 
        for(int j=1;j<=n;++j)
            if(f[j][i-1]) 
                f[j][i]=f[f[j][i-1]][i-1];

    for(int i=1;i<=m;++i)
    {
        int x,y;

        cin>>x>>y;
        cout<<dist[x]+dist[y]-2*dist[LCA(x, y)]<<endl;
    }

    return 0;
}