ural 1471 Tree（最近公共祖先）

裸LCA，用的DFS+并查集，这题弄了我一下午。。。。

#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<queue>
#include<stack>
#include<map>
#include<vector>

using namespace std;
const int maxn=50010;
const int maxm=75010;

struct E
{
	int u,v,w,next;	
}edge[maxn*4];
struct query
{
	int u,v,id,next;	
}q[maxm*2];
int dist[maxn],head[maxn],hq[maxn],res[maxm][4],f[maxn];
bool vis[maxn];
int n,m,e;

void addedge(int u,int v,int w)
{
	edge[e].u=u;
	edge[e].v=v;
	edge[e].w=w;
	edge[e].next=head[u];
	head[u]=e++;	
}
void addq(int u,int v,int id)
{
	q[e].u=u;	
	q[e].v=v;
	q[e].id=id;
	q[e].next=hq[u];
	hq[u]=e++;
}
int find(int x) { return x==f[x]?x:find(f[x]); }

void LCA(int s)
{
	vis[s]=1;
	f[s]=s;
	for(int i=hq[s];i!=-1;i=q[i].next)
	{
		int v=q[i].v;
		if(vis[v]) res[q[i].id][2]=find(v);	
	}
	for(int i=head[s];i!=-1;i=edge[i].next)
	{
		int v=edge[i].v;
		if(!vis[v])
		{
			dist[v]=dist[s]+edge[i].w;
			LCA(v);
			f[v]=s;	
		}
	}
}

int main()
{
	int a,b,c;
	while(~scanf("%d",&n))
	{
		memset(head,-1,sizeof(head));
		memset(hq,-1,sizeof(hq));	
		memset(vis,0,sizeof(vis));
		memset(dist,0,sizeof(dist));
		for(int i=0;i<=n;i++)
			f[i]=i;
		e=0;
		for(int i=1;i<n;i++)
		{
			scanf("%d%d%d",&a,&b,&c);	
			addedge(a,b,c);
			addedge(b,a,c);
		}
		e=0;
		scanf("%d",&m);
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d",&a,&b);	
			addq(a,b,i);
			addq(b,a,i);
			res[i][0]=a,res[i][1]=b;
		}
		dist[1]=0;
		LCA(1);
		for(int i=1;i<=m;i++)
		{
			printf("%d\n",dist[res[i][0]]+dist[res[i][1]]-2*dist[res[i][2]]);	
		}
	}
    return 0;
}