C++ 最近公共祖先LCA

#include<bits/stdc++.h>
using namespace std;
struct node{
	int v,w,nxt;
}edge[100010];
int n,m,u,v,w;
int cnt;//边的编号
int head[50010];//head[i]表示i节点最后一条边的编号
void add(int u,int v,int w){
	cnt++;
	edge[cnt].v=v;
	edge[cnt].w=w;
	edge[cnt].nxt=head[u];
	head[u]=cnt;
} 
int dep[50010];//每个节点的深度
int dis[50010];//每个节点到根节点的距离
int p[50010][20];//记录上跳2^k后的祖先节点数组 
void dfs(int u,int fa){
	dep[u]=dep[fa]+1;
	p[u][0]=fa;//u向上跳2^0步，到达自己的父节点fa
	for(int i=head[u];i;i=edge[i].nxt){
		int v=edge[i].v;
		int w=edge[i].w;
		if(v==fa) continue;
		dis[v]=dis[u]+w;
		dfs(v,u);
	} 
}
int lca(int x,int y){
	if(dep[x]<dep[y]){ //保证x在下面 
		swap(x,y);
	}
	//将x上跳到和y相同的深度
	for(int i=19;i>=0;i--){
		if(dep[x]-(1<<i) >= dep[y]){
			x=p[x][i];
		}
	} 
	//特判
	if(x==y) return x;
	//x和y一起上跳
	for(int i=19;i>=0;i--){
		if(p[x][i]!=p[y][i]){
			x=p[x][i];
			y=p[y][i];
		}
	}  
	return p[x][0];
}
int main(){
	cin>>n>>m;
	for(int i=1;i<n;i++){ //n个节点的树有n-1条边 
		cin>>u>>v>>w;
		add(u,v,w);
		add(v,u,w);
	}
	dfs(1,0); 
	for(int j=1;j<20;j++){ //处理每一个节点的p数组 
		for(int i=1;i<=n;i++){
			p[i][j]=p[p[i][j-1]][j-1];
		}
	} 
	//LCA
	for(int i=1;i<=m;i++){
		cin>>u>>v;
		int lCA=lca(u,v);
		cout<<dis[u]-dis[lCA]+dis[v]-dis[lCA]<<endl;
	}
	return 0;
}