题解:
树上倍增LCA处理最近公共祖先,并查集处理不同集合的树,递推距离,但是有一个坑点,这题卡常,卡的要命,所以在处理一个集合的树的时候,需要剪枝,访问过的点,就不需要再访问了,但是一般建树的dfs,访问过的点,可能还会访问一两次。
AC代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int base1=131;
const ll maxn=1e5+5;
struct node
{
ll to,nex,w;
} edge[maxn<<1];
ll head[maxn],cnt,deep[maxn],pre[maxn][30];
ll fa[maxn],vis[maxn],dis[maxn];
ll fin(ll x)
{
if(x!=fa[x])
{
fa[x]=fin(fa[x]);
}
return fa[x];
}
void join(ll x,ll y)
{
x=fin(x);
y=fin(y);
if(x!=y)
fa[x]=y;
}
void add(ll u,ll v,ll w)
{
edge[cnt].to=v;
edge[cnt].w=w;
edge[cnt].nex=head[u];
head[u]=cnt++;
}
void init()
{
memset(vis,0,sizeof(vis));
memset(dis,0,sizeof(dis));
memset(head,-1,sizeof(head));
memset(pre,0,sizeof(pre));
memset(fa,0,sizeof(fa));
}
void dfs(ll u,ll fa)
{
vis[u]=1;
deep[u]=deep[fa]+1;
pre[u][0]=fa;
for(ll i=1; (1<<i)<=deep[u]; i++)//向上倍增
pre[u][i]=pre[pre[u][i-1]][i-1];
for(ll i=head[u]; ~i; i=edge[i].nex)
{
ll v=edge[i].to;
if(v!=fa&&!vis[v])
{
dis[v]=dis[u]+edge[i].w;
dfs(v,u);
}
}
}
ll lca(ll u,ll v)
{
if(deep[u]<deep[v])
swap(u,v);
ll i=-1;
while(deep[u]-(1<<(i+1))>=0)
i++;
for(ll j=i; j>=0; j--)
if(deep[u]-(1<<j)>=deep[v])
u=pre[u][j];
if(u==v)
return u;
for(ll j=i; j>=0; j--)
if(pre[u][j]!=pre[v][j])
u=pre[u][j],v=pre[v][j];
return pre[u][0];
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
ll n,m,s;
while(cin>>n>>m>>s)
{
init();
cnt=0;
for(ll i=1; i<=n; i++)
fa[i]=i;
for(ll i=1; i<=m; i++)
{
ll u,v,w;
cin>>u>>v>>w;
join(u,v);
add(u,v,w);
add(v,u,w);
}
for(ll i=1; i<=n; i++)
{
if(!vis[i])
{
dfs(i,0);
}
}
while(s--)
{
ll x,y;
cin>>x>>y;
if(fin(x)!=fin(y))
{
cout<<"Not connected"<<endl;
}
else
{
ll lc=lca(x,y);
cout<<dis[x]+dis[y]-2*dis[lc]<<endl;
}
}
}
}