hdu2586 How far away ？

hdu2586
题意呢就是说有一棵树，求两点之间的距离
就算求出lca，然后算每个点到根的距离，结果就算两个点到根的距离之和减去最近公共祖先到根的距离
注意：多组测试数据！！！

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
#define N 44000
struct node{
    int x,y,z,next;
}map[2*N];
int num,f[N][20],sum[N],h[N],lg[N],n,m,dep[N],T;
void insert(int x,int y,int z){
    map[++num].x=x;map[num].y=y;map[num].z=z;map[num].next=h[x];h[x]=num;
    map[++num].x=y;map[num].y=x;map[num].z=z;map[num].next=h[y];h[y]=num;
}
void dfs(int u,int s){
    for(int i=1;i<=lg[n];i++){
        if(!f[u][i-1]) break;
        f[u][i]=f[f[u][i-1]][i-1];
    }
    for(int i=h[u];i;i=map[i].next){
        int v=map[i].y;
        if(!dep[v]){
            dep[v]=dep[u]+1;f[v][0]=u;sum[v]=sum[u]+map[i].z;
            dfs(v,sum[v]);
        }
    } 
} 
int lca(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    int len=dep[x]-dep[y];
    for(int i=0;i<=lg[len];i++)
        if(len>>i&1) x=f[x][i]; 
    if(x==y) return x;
    for(int i=lg[n];i>=0;i--)
        if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
int main(){
    scanf("%d",&T);
    lg[0]=-1;for(int i=1;i<=40000;i++) lg[i]=lg[i>>1]+1;
    while(T--){
        scanf("%d%d",&n,&m);num=0;
        memset(h,0,sizeof(h));
        memset(sum,0,sizeof(sum));
        memset(dep,0,sizeof(dep));dep[1]=1;
        for(int i=1;i<n;i++){
            int x,y,z;scanf("%d%d%d",&x,&y,&z);
            insert(x,y,z); 
        }dfs(1,0);
        for(int i=1;i<=m;i++){
            int x,y;scanf("%d %d",&x,&y);
            printf("%d\n",sum[x]+sum[y]-2*sum[lca(x,y)]);
        }
    }
    return 0;
}