题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2586
题意:一个村庄有n个房子和n-1条双向路,每两个房子之间都有一条简单路径。
现在有m次询问,求两房子之间的距离
由题意可知,可用其中一个点作为根节点建树,求两点距离转化为
求LCA:最近公共祖先,可以用tarjan的思想搜索子树回溯+并查集离线求解
两点距离=根节点到两点距离之和-根节点到LCA距离*2
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define N 40100
#define M 210
struct node{
int to,w,next;
}edge[N*2];
int n,m; //点数,询问次数
int head[N];
int k;
int fa[N]; //父亲结点
int dis[N]; //到根节点距离
int vis[N]; //是否访问过
int s[M]; //询问起点
int e[M]; //询问终点
int lca[M]; //LCA(s,e) 最近公共祖先
int find(int x){
if(fa[x]!=x) return fa[x]=find(fa[x]);
return fa[x];
}
void init(){
k=1;
memset(head,0,sizeof(head));
memset(dis,0,sizeof(dis));
memset(vis,0,sizeof(vis));
}
void add(int u,int v,int w){
edge[k].to=v;
edge[k].w=w;
edge[k].next=head[u];
head[u]=k++;
}
void tarjan(int u){
int i,v;
fa[u]=u;
vis[u]=1;
for(i=0;i<m;++i){
if(e[i]==u&&vis[s[i]])lca[i]=find(s[i]); //若询问的两点中有一点已被访问过,则两点的LCA则为这一点的当前父节点
if(s[i]==u&&vis[e[i]])lca[i]=find(e[i]);
}
for(i=head[u];i;i=edge[i].next){
v=edge[i].to;
if(!vis[v]){//若没被访问过
dis[v]=dis[u]+edge[i].w;//更新距离
tarjan(v);
fa[v]=u;//回溯更新父节点
}
}
}
int main(){
int t;
int i,j;
int a,b,c;
scanf("%d",&t);
while(t--){
init();
scanf("%d%d",&n,&m);
for(i=0;i<n-1;++i){
scanf("%d%d%d",&a,&b,&c);
add(a,b,c);
add(b,a,c);
}
for(i=0;i<m;++i){
scanf("%d%d",&s[i],&e[i]);
}
tarjan(1);
for(i=0;i<m;++i){
printf("%d\n",dis[s[i]]+dis[e[i]]-2*dis[lca[i]]);//两点距离为根节点到两点距离之和-根节点到LCA距离*2
}
}
return 0;
}