HDU 2586 How far away ？ LCA模板题

题目：勇气小镇是一个有着n个房屋的小镇，为什么把它叫做勇气小镇呢，这个故事就要从勇气小镇成立的那天说起了，
修建小镇的时候，为了让小镇有特色，镇长特地只修了n-1条路，并且规定说，所有在勇气小镇的村民，每一次出门必须规划好路线,
路线必须满足在到达终点之前绝对不走回头路。每个人都要这样，不然那个人就不配在小镇生活下去，因为他没有这个勇气。
事实上，这并不能算一项挑战，因为n-1条路已经连通了每户人家，不回头地从起点到终点，只是一个时间上的问题。
由于小镇上的福利特别好，所以小懒入住了这个小镇，他规划了m次的行程，每次从L房屋到R房屋,他想问你他每次从L房屋到R房屋需要走多远的路。

lca模板题，第一次写lca,这是参考博客，传送门

AC代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <cmath>
#include <algorithm>
using namespace std;
const int maxn=1e5+5;
struct node {
	int v,w,nxt;
} e[maxn];
int head[maxn],tot,vis[maxn],dep[maxn],d[maxn],f[maxn][20];
int t,lg,n,m,u,v,w;
queue<int>q;
void init() {
	tot=1;
	lg=int(log(n)/log(2))+1;
	memset(vis,0,sizeof(vis));
	memset(d,0,sizeof(d));
	memset(f,0,sizeof(f));
	memset(dep,0,sizeof(dep));
	memset(head,-1,sizeof(head));
}
void add(int u,int v,int w) {
	e[tot].v=v;
	e[tot].w=w;
	e[tot].nxt=head[u];
	head[u]=tot++;
}
void bfs() {
	dep[1]=1;
	q.push(1);
	while(!q.empty()) {
		int u=q.front();
		q.pop();
		for(int i=head[u]; i!=-1; i=e[i].nxt) {
			int v=e[i].v;
			if(dep[v])continue;//深度已经处理过，则跳过该点，避免返回上一层
			dep[v]=dep[u]+1;
			d[v]=d[u]+e[i].w;
			f[v][0]=u;//向上跳2^0=1步即是父节点
			for(int k=1; k<=lg; k++)f[v][k]=f[f[v][k-1]][k-1]; //v向上跳2^k步等价于v跳2^(k-1)步后再跳2^(k-1)步
			q.push(v);
		}
	}
}
int lca(int x,int y) {
	if(dep[y]>dep[x])swap(x,y);//保证x的深度大于y，方便下面判断
	for(int i=lg; i>=0; i--) { //二进制优化（倍增）由大到小跳
		if(dep[f[x][i]]>=dep[y])x=f[x][i];//跳后深度仍然不大于y说明可以跳
		}
	if(x==y)return x;
	for(int i=lg; i>=0; i--) { //当x不能再跳了，那就让x和y一起跳
		if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];//相等那就是祖先，不能跳
	}
}
int main() {
	scanf("%d",&t);
	while(t--) {
		scanf("%d%d",&n,&m);
		init();
		for(int i=1; i<n; i++) {
			scanf("%d%d%d",&u,&v,&w);
			add(u,v,w);
			add(v,u,w);
		}
		bfs(); //确定深度 与 1点到其他各点的距离
		for (int i = 1; i <= m; i++) {
			scanf("%d%d", &u, &v);
			printf("%d\n", d[u] + d[v] - 2 * d[lca(u, v)]);
		}
	}
}