hdu2586 how far away?(lca求距离）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
struct node{
	int to,nex,wei;
}edge[200005];
int head[100005],cnt,dis[100005],dep[100005],f[100005][20];
void add(int u,int v,int w){
	edge[cnt].to=v;
	edge[cnt].nex=head[u];
	edge[cnt].wei=w;
	head[u]=cnt++;
}
void dfs(int root,int d,int val,int fa){//根，深度，权值，父亲
	dis[root]=val;
        f[d][0]=fa;
	dep[root]=d;
	for(int i=head[root];~i;i=edge[i].nex){
		int v=edge[i].to;
		if(v==fa)continue;
		dfs(v,d+1,val+edge[i].wei,root);
	}
}
int lca(int x,int y){
    if(dep[x]<dep[y])swap(x,y);
//令x为深度较深的点
    for(int i=16;i>=0;i--)if(dep[f[x][i]]>=dep[y])x=f[x][i];
//让x向上走到与y同一深度
    if(x==y)return x; //如果直接是lca，直接返回
    for(int i=16;i>=0;i--)if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
//x，y同时向上走，直到父节点相同
    return f[x][0]; //返回父节点
}
int main(){
	int t;
	scanf("%d",&t);
	while(t--){
		int n,m,i,j;
		scanf("%d%d",&n,&m);
		memset(head,-1,sizeof(head));
		cnt=0;
		for(i=1;i<n;i++){
			int u,v,w;
			scanf("%d%d%d",&u,&v,&w);
			add(u,v,w);
			add(v,u,w);
		}
		dfs(1,1,0,1);
		for(int k=1;k<=16;k++){
			for(i=1;i<=n;i++){
				f[i][k]=f[f[i][k-1]][k-1];
			}
		}
		for(i=1;i<=m;i++){
			int u,v;
			scanf("%d%d",&u,&v);
			int ls=lca(u,v);
			printf("%d\n",dis[u]+dis[v]-2*dis[ls]);
		}
	}
	return 0;
}