【BZOJ 1832】 [AHOI2008]聚会|倍增lca

100ac 留念

双倍经验题

两两求距离 相加除以二为距离

两两求lca depth最大为三个的lca

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN=500000+10;
int to[MAXN][20],depth[MAXN];
int tot,g[MAXN],nnext[MAXN*2],num[MAXN*2] ;
int team[MAXN],head,tail;
void add(int x,int y)
{
tot++;
nnext[tot]=g[x];
g[x]=tot;
num[tot]=y;
}
void bfs()
{
depth[1]=1;
team[++tail]=1;
while(head<tail)
{
int x=team[++head];
for(int i=1;i<=19;i++) to[x][i]=to[to[x][i-1]][i-1];
for(int i=g[x];i;i=nnext[i])
{
int tmp=num[i];
if(tmp==to[x][0]) continue;
depth[tmp]=depth[x]+1;
to[tmp][0]=x;
team[++tail]=tmp;
}
}
}
int lca(int x,int y)
{
if(depth[x]<depth[y])swap(x,y);
for(int i=19;i>=0;i--)
if(depth[to[x][i]]>=depth[y])
x=to[x][i];
if(x==y) return x;
for(int i=19;i>=0;i--)
if(to[x][i]!=to[y][i])
x=to[x][i],y=to[y][i];
return to[x][0];
}
int dis(int x,int y,int fa)
{
return abs(depth[x]-depth[fa])+abs(depth[y]-depth[fa]);
}
int main()
{
int n,m;
int x,y,z;
scanf("%d %d",&n,&m);
for(int i=1;i<n;i++) scanf("%d %d",&x,&y),add(x,y),add(y,x);
bfs();
for(int i=1;i<=m;i++)
{
scanf("%d %d %d",&x,&y,&z);
int zx=lca(x,z);
int zy=lca(z,y);
int xy=lca(x,y);
int ans=zx;
if(depth[ans]<depth[zy]) ans=zy;
if(depth[ans]<depth[xy]) ans=xy;
int tmp1=dis(x,y,xy);
int tmp2=(dis(x,z,zx)+dis(z,y,zy)-tmp1)/2;
printf("%d %d\n",ans,tmp1+tmp2);
}
return 0;
}