HDU2586【LCA在线算法】

最新推荐文章于 2019-08-21 18:35:43 发布

keyboarder_zsq

最新推荐文章于 2019-08-21 18:35:43 发布

阅读量400

点赞数

分类专栏： LCA 文章标签： ACM

本文链接：https://blog.csdn.net/KEYboarderQQ/article/details/70187207

版权

LCA 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

求树上两两之间距离。
思路：

在一棵树上，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;
}