求树上两两之间距离。
思路:
在一棵树上,dis[ index ]代表root到index的距离,u,v 的最短距离那不就是:dis[ u ] + dis[ v ] - 2 * dis[ LCA( u , v ) ];
然后这边可能有多棵树,先利用并查集并一下,然后拿 结点0 ,把所有的根连起来,DFS,ST表,然后RMQ,就可以在线查询LCA了。
如过LCA( u , v ) = 0,没有;
否则:dis[ u ] + dis[ v ] - 2 * dis[ LCA( u , v ) ];
裸的LCA。
//#include <bits/stdc++.h>
//using namespace std;
#include<cstdio>
#include<cstring>
#include<algorithm>
typedef long long LL;
//#pragma comment(linker, "/STACK:102400000,102400000")
const int N=4e4+10;
struct Edge{
int w;
int to;
int next;
}e[N<<1];
int head[N],tol;
int n;
void add(int u,int v,int w)
{
e[tol].w = w;
e[tol].to = v;
e[tol].next = head[u];
head[u] = tol++;
}
int pre[N];
int Find(int x)
{
int r = x;
while(pre[r] != r)
r = pre[r];
int i = x, j;
while(pre[i] != r)
{
j = pre[i];
pre[i] = r;
i = j;
}
return r;
}
void init(){
tol=0;
memset(head,-1,sizeof(head));
for(int i=1;i<=n;++i) pre[i]=i;
}
int dis[N];
bool vis[N];
int deep[N<<1],first[N],tot,id[N<<1];
void DFS(int u,int dep,int sum)
{
dis[u] = sum;
vis[u] = true;
id[++tot] = u;
deep[tot] = dep;
first[u] = tot;
for(int i=head[u];~i;i=e[i].next)
{
int v=e[i].to;
if(vis[v]) continue;
DFS(v,dep+1,sum + e[i].w);
id[++tot] = u;
deep[tot] = dep;
}
}
int dp[N<<1][25];
void ST()
{
for(int i=1; i<=tot; i++) dp[i][0]=i;
for(int j=1; (1<<j)<=tot; j++){
for(int i=1; i+(1<<j)-1<=tot; i++){
int a=dp[i][j-1],b=dp[i+(1<<(j-1))][j-1];
dp[i][j]=deep[a] < deep[b]?a:b;
}
}
}
int RMQ(int Left,int Right)
{
int k=0;
while((1<<(k+1)) <= (Right-Left+1)) k++;
int a=dp[Left][k],b=dp[Right-(1<<k)+1][k];
return deep[a]<deep[b]?a:b;
}
int LCA(int u,int v)
{
int x=first[u],y=first[v];
if(x>y){
/*********///莫名奇妙这里打错了,T了。。。。
x = x^y;
y = x^y;
x = x^y;
/*************/
}
int res=RMQ(x,y);
return id[res];
}
int main()
{
int m,q;
int u,v,w;
while(~scanf("%d%d%d",&n,&m,&q))
{
init();
while(m--){
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);
add(v,u,w);
}
for(int i=1;i<=n;i++)
if(pre[i] == i)
{
add(0,i,0);
add(i,0,0);
}
tot = 0;
memset(vis,false,sizeof(vis));
memset(dis,0,sizeof(dis));
DFS(0,1,0);
ST();
while(q--)
{
scanf("%d%d",&u,&v);
int uv = LCA(u,v);
if(!uv)
puts("Not connected");
else
printf("%d\n",dis[u] + dis[v] - 2 * dis[uv]);
}
}
return 0;
}