题目概述
给出一棵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;
}