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;}