HDU 2874 Connections between cities (樹上倍增 LCA + 並查集 ）

Connections between cities:

題解：

樹上倍增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;
            }
        }
    }
}