【LCA】HDU2586[How far away?]题解

题目概述

给出一棵n个节点的树和m个询问，每个询问形如x,y表示求x,y的距离。

解题报告

题目很裸，这篇文章的意义就是水博客放一下LCA模板啦。

示例程序

ST算法。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=40000,Log=15;

int te,n,Q,dep[maxn+5],dis[maxn+5],fa[maxn+5][Log+5];
int E,lnk[maxn+5],son[2*maxn+5],nxt[2*maxn+5],w[2*maxn+5];

void Add(int x,int y,int z) {son[++E]=y;w[E]=z;nxt[E]=lnk[x];lnk[x]=E;}
void Dfs(int x)
{
    for (int j=lnk[x];j;j=nxt[j]) if (son[j]!=fa[x][0])
        fa[son[j]][0]=x,dis[son[j]]=dis[x]+w[j],dep[son[j]]=dep[x]+1,Dfs(son[j]);
}
void make_fa()
{
    for (int j=1;j<=Log;j++)
    for (int i=1;i<=n;i++)
        fa[i][j]=fa[fa[i][j-1]][j-1];
}
int LCA(int x,int y)
{
    if (dep[x]<dep[y]) swap(x,y);
    for (int i=Log;i>=0;i--) if (dep[fa[x][i]]>=dep[y]) x=fa[x][i];
    if (x==y) return x;
    for (int i=Log;i>=0;i--) if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int main()
{
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    scanf("%d",&te);
    while (te--)
    {
        E=0;memset(lnk,0,sizeof(lnk));
        scanf("%d%d",&n,&Q);
        for (int i=1,x,y,z;i<=n-1;i++)
            scanf("%d%d%d",&x,&y,&z),Add(x,y,z),Add(y,x,z);
        fa[1][0]=1;Dfs(1);make_fa();
        while (Q--)
        {
            int x,y;scanf("%d%d",&x,&y);int lca=LCA(x,y);
            printf("%d\n",dis[x]-dis[lca]+dis[y]-dis[lca]);
        }
    }
    return 0;
}

Tarjan_LCA。

#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=40000,maxm=200,maxe=2*(maxn+maxm);

int te,n,Q,dis[maxn+5],father[maxn+5],ans[maxm+5];
int E,lnk[maxn+5][2],son[maxe+5],nxt[maxe+5],w[maxe+5];
bool vis[maxn+5];

void Add(int x,int y,int z,int ID) {son[++E]=y;w[E]=z;nxt[E]=lnk[x][ID];lnk[x][ID]=E;}
int getfa(int x)
{
    if (father[x]==x) return x;
    return father[x]=getfa(father[x]);
}
void Tarjan_LCA(int x)
{
    vis[x]=true;father[x]=x;
    for (int j=lnk[x][0];j;j=nxt[j]) if (!vis[son[j]])
        dis[son[j]]=dis[x]+w[j],Tarjan_LCA(son[j]),father[son[j]]=x;
    for (int j=lnk[x][1];j;j=nxt[j])
        if (vis[son[j]]) ans[w[j]]=dis[x]-dis[getfa(son[j])]+dis[son[j]]-dis[getfa(son[j])];
}
int main()
{
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    scanf("%d",&te);
    while (te--)
    {
        E=0;memset(lnk,0,sizeof(lnk));memset(vis,0,sizeof(vis));
        scanf("%d%d",&n,&Q);
        for (int i=1,x,y,z;i<=n-1;i++)
            scanf("%d%d%d",&x,&y,&z),Add(x,y,z,0),Add(y,x,z,0);
        for (int i=1;i<=Q;i++)
        {
            int x,y;scanf("%d%d",&x,&y);
            Add(x,y,i,1);Add(y,x,i,1);
        }
        Tarjan_LCA(1);
        for (int i=1;i<=Q;i++) printf("%d\n",ans[i]);
    }
    return 0;
}