题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2586
大致题意:给你一棵树,树上有n个点,n-1条边,有q次查询,每次查询树上任意两点间的距离。
题解:LCA经典题,定义一个dis数组表示该点到树根(假如有根的话)的距离,则根据LCA中的
Tarjan算法的工作机制和原理,保存查询的两点,之后跑一发Tarjan,有关系(其中待求的两点)
的两点,假如另一点已经访问,则找到这两点的最近公共祖先,很容易推出,这两点间的距离就是
dfs[ i ]+dis[ j ]-2*dis[ z[i] ];其中z[i]为i和j的最近公共祖先。
#include<map>
#include<stack>
#include<queue>
#include<vector>
#include<math.h>
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
using namespace std;
typedef long long ll;
#define inf 1000000000
#define mod 1000000007
#define maxn 40010
#define lowbit(x) (x&-x)
#define eps 1e-10
int dis[maxn],vis[maxn],n,m,e1[maxn],e2[maxn],fa[maxn],e3[maxn];
//e1和e2数组分别存待求的两点,e3存这两点的最近公共祖先
struct node
{
int id,c;
};
vector<node>q[maxn];
int find(int x)
{
if(fa[x]==x)
return x;
return fa[x]=find(fa[x]);
}
void tarjan(int x)
{
int i;
vis[x]=1;fa[x]=x;
for(i=1;i<=m;i++)
{
if(e1[i]==x && vis[e2[i]])
e3[i]=find(e2[i]);
if(e2[i]==x && vis[e1[i]])
e3[i]=find(e1[i]);
}
for(i=0;i<q[x].size();i++)
{
node v=q[x][i];
if(vis[v.id])
continue;
dis[v.id]=dis[x]+v.c;
tarjan(v.id);
fa[v.id]=x;
}
}
int main(void)
{
node tmp;
int T,i,x,y,z;
scanf("%d",&T);
while(T--)
{
memset(e1,0,sizeof(e1));
memset(e2,0,sizeof(e2));
memset(e3,0,sizeof(e3));
memset(vis,0,sizeof(vis));
memset(dis,0,sizeof(dis));
scanf("%d%d",&n,&m);
for(i=1;i<n;i++)
{
scanf("%d%d%d",&x,&y,&z);
tmp.id=x;tmp.c=z;
q[y].push_back(tmp);
tmp.id=y;tmp.c=z;
q[x].push_back(tmp);
}
for(i=1;i<=m;i++)
scanf("%d%d",&e1[i],&e2[i]);
dis[1]=0;//默认1为树根
tarjan(1);
for(i=1;i<=m;i++)
printf("%d\n",dis[e1[i]]+dis[e2[i]]-2*dis[e3[i]]);
for(i=0;i<=maxn;i++)
if(q[i].empty()==0)
q[i].clear();
}
return 0;
}