HDU2586【LCA在线算法】

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

在一棵树上,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 
//using namespace std;
#include
#include
#include
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<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;
}



你可能感兴趣的:(LCA)