【LCA最近公共祖先】HDU 2586 How far away ？

来源：互联网发布：刷流量软件编辑：程序博客网时间：2024/06/12 01:23

题目链接：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 210struct 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;}

0 0