背景:
发现树剖还有好多题没有做。
题目传送门:
https://www.luogu.org/problemnew/show/P4281
题意:
多组询问,在树上找一个点,使给定的三个点到这个点之间的距离和最小。
思路:
我们知道树上任意三点两两的
L
C
A
LCA
LCA共三个
t
1
=
l
c
a
(
x
,
y
)
,
t
2
=
l
c
a
(
x
,
z
)
,
t
3
=
l
c
a
(
y
,
z
)
t1=lca(x,y),t2=lca(x,z),t3=lca(y,z)
t1=lca(x,y),t2=lca(x,z),t3=lca(y,z),且至少存在一组相等。
证明如下:
设
T
=
l
c
a
(
x
,
y
,
z
)
T=lca(x,y,z)
T=lca(x,y,z),即三个点的
L
C
A
LCA
LCA,那么必然存在
l
c
a
(
t
1
,
t
2
)
=
l
c
a
(
t
1
,
t
3
)
=
l
c
a
(
t
2
,
t
3
)
=
T
lca(t1,t2)=lca(t1,t3)=lca(t2,t3)=T
lca(t1,t2)=lca(t1,t3)=lca(t2,t3)=T。
得证。
因此我们只需找出不相等的那一个作为目标节点即可。
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,len=0,ans=0;
int last[500010],dep[500010],f[500010][25];
struct node{int x,y,next;} a[1000010];
void ins(int x,int y)
{
a[++len]=(node){x,y,last[x]}; last[x]=len;
}
void init()
{
for(int j=1;j<=20;j++)
for(int i=1;i<=n;i++)
f[i][j]=f[f[i][j-1]][j-1];
}
void dfs(int x)
{
for(int i=last[x];i;i=a[i].next)
{
int y=a[i].y;
if(y==f[x][0]) continue;
dep[y]=dep[x]+1;
f[y][0]=x;
dfs(y);
}
}
int lca(int x,int y)
{
if(dep[x]<dep[y]) swap(x,y);
for(int i=20;i>=0;i--)
if(dep[y]<=dep[f[x][i]]) x=f[x][i];
if(x==y) return x;
for(int i=20;i>=0;i--)
if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
return f[x][0];
}
int main()
{
int x,y,z;
scanf("%d %d",&n,&m);
for(int i=1;i<n;i++)
{
scanf("%d %d",&x,&y);
ins(x,y),ins(y,x);
}
dep[1]=1;
dfs(1);
init();
for(int i=1;i<=m;i++)
{
scanf("%d %d %d",&x,&y,&z);
int t1=lca(x,y),t2=lca(x,z),t3=lca(y,z);
if(t1==t2) printf("%d ",t3);
else if(t1==t3) printf("%d ",t2);
else printf("%d ",t1);
printf("%d\n",dep[x]+dep[y]+dep[z]-dep[t1]-dep[t2]-dep[t3]);
}
}