版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/monochrome00/article/details/82594623
题目链接<http://acm.hdu.edu.cn/showproblem.php?pid=2874>
题意:
给出一个森林,要求判断两个点是否在同一棵树上,如果在输出两者之间的简单路径长度。
题解:
关于LCA:https://www.cnblogs.com/FuTaimeng/p/5655616.html
是否在同一棵树上就用并查集判断一下。
树上两点间距就是根节点到两点间距之和,减去根节点到最近公共祖先距离的两倍。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e4+7;
int fa[N][20],d[N],dis[N];
int p[N],edn;
int n,m,c;
struct Edge{
int u,v,w,nxt;
Edge(int u=0,int v=0,int w=0,int nxt=0):u(u),v(v),w(w),nxt(nxt){}
}edge[N*2];
void add(int u,int v,int w){
edge[++edn]=Edge(u,v,w,p[u]);p[u]=edn;
edge[++edn]=Edge(v,u,w,p[v]);p[v]=edn;
}
void dfs(int u){
for(int i=p[u];~i;i=edge[i].nxt){
int v=edge[i].v;
if(v==fa[u][0])continue;
d[v]=d[u]+1;
dis[v]=dis[u]+edge[i].w;
fa[v][0]=u;
dfs(v);
}
}
void init(){
for(int j=1;(1<<j)<=n;j++){
for(int i=1;i<=n;i++){
fa[i][j]=fa[fa[i][j-1]][j-1];
}
}
}
int lca(int a,int b){
if(d[a]>d[b])swap(a,b);
int f=d[b]-d[a];
for(int i=0;(1<<i)<=f;i++){
if((1<<i)&f) b=fa[b][i];
}
if(a!=b){
for(int i=(int)log2(N);i>=0;i--){
if(fa[a][i]!=fa[b][i]){
a=fa[a][i]; b=fa[b][i];
}
}
a=fa[a][0];
}
return a;
}
int f[N];
int fd(int x){return x==f[x]?x:f[x]=fd(f[x]);}
int main()
{
while(scanf("%d%d%d",&n,&m,&c)!=EOF){
memset(p,-1,sizeof(p)),edn=-1;
memset(d,0,sizeof(d));
memset(dis,0,sizeof(dis));
for(int i=1;i<=n;i++) f[i]=i;
int u,v,w;
for(int i=1;i<=m;i++){
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);
int fu=fd(u),fv=fd(v);
f[fu]=fv;
}
for(int i=1;i<=n;i++){
if(i==fd(i)) dfs(i);
}
init();
for(int i=1;i<=c;i++){
scanf("%d%d",&u,&v);
int fu=fd(u),fv=fd(v);
if(fu==fv){
int pp=lca(u,v);
printf("%d\n",dis[u]+dis[v]-2*dis[pp]);
}
else printf("Not connected\n");
}
}
}