HDU 2586 How far away ？(最长公共祖先 LCA模板题)

#include <bits/stdc++.h>
using namespace std;

const int maxn=40005;
int f[maxn],head[maxn],dis[maxn],ans[202],vis[maxn];
int tot,n;
struct Edge{
	int u,v,w,next;
};
Edge edge[maxn*2];
vector< pair<int,int> > Q[maxn];
int find(int x)
{
	return f[x]==x?x:x=find(f[x]);
}
void add(int u,int v,int w)
{
	edge[tot].u=u;
	edge[tot].v=v;
	edge[tot].w=w;
	edge[tot].next=head[u];
	head[u]=tot++;
}
void init()
{
	for(int i=0;i<=maxn;i++)
	Q[i].clear(),f[i]=i,ans[i]=0,dis[i]=0,vis[maxn],head[i]=-1;
	tot=0;
}
void dfs(int u,int d)
{
	dis[u]=d;
	vis[u]=1;
	for(int i=head[u];~i;i=edge[i].next)
	{
		int v=edge[i].v;
		int w=edge[i].w;
		if(!vis[v])
		{
			dfs(v,d+w);
			f[v]=u;
		}
	}
	
	for(int i=0;i<Q[u].size();i++)
	{
		int v=Q[u][i].first;
		int id=Q[u][i].second;
		if(vis[v])
		{
			int lca=find(v);
			ans[id]=dis[u]-dis[lca]+dis[v]-dis[lca];
		}
	}
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		init();
		int q;
		scanf("%d %d",&n,&q);
		for(int i=0;i<n-1;i++)
		{
			int u,v,w;
			scanf("%d %d %d",&u,&v,&w);
			add(u,v,w);
			add(v,u,w);
		}
		for(int i=0;i<q;i++)
		{
			int u,v;
			scanf("%d %d",&u,&v);
			Q[u].push_back(make_pair(v,i));
			Q[v].push_back(make_pair(u,i));//
		}
		dfs(1,0);
		for(int i=0;i<q;i++)printf("%d\n",ans[i]);
	}
	return 0;
}