在线倍增算法求LCA

给定一棵树，询问两个节点的中点。

A[x,i]代表与x深度相差2的i次方的祖先。A满足性质

A[x,i+1]=A[A[x,i],i]

令j是使2的j次方小于等于H的最大非负整数

Ancestor(x,H)=Ancestor(A[x,j],H-2的j次方)

则LCA(x,y)=LCA(Ancestor(x,deep(y)-deep(x)),y)

#include<cstdio>
#include<cstring>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<math.h>
#include<vector>
using namespace std;

vector<int>g[10001];

int A[10001][15];
int d[10001];

void dfs(int root,int deep,int p)
{
d[root]=deep;
A[root][0]=p;
for(int i=1;i<15;i++)
{
A[root][i]=A[A[root][i-1]][i-1];
}
for(int i=0;i<g[root].size();i++)
{
if(!d[g[root][i]])

dfs(g[root][i],deep+1,root);
}
}

void BuildTree(int N)
{
int a,b;
memset(A,0,sizeof(A));
memset(d,0,sizeof(d));
for(int i=1;i<=N;i++)g[i].clear();
for(int i=1;i<N;i++)
{
scanf("%d%d",&a,&b);
g[a].push_back(b);
g[b].push_back(a);
}
dfs(1,1,0);
}

int Ancestor(int x,int len)
{
for(int i=14;i>=0;i--)
{
if(len >= 1<<i)
{
len -= 1<<i;
x=A[x][i];
}
}
return x;
}

int LCA(int x,int y)
{
if(d[x]>d[y])
{
x=Ancestor(x,d[x]-d[y]);
}
else
y=Ancestor(y,d[y]-d[x]);
if(x==y)
return x;
int ret;
int i=14;
while(i--)
{
if(A[x][i]==A[y][i])
{
ret=A[x][i];
}
else
{
x=A[x][i];
y=A[y][i];
}
}
return ret;
}

void Answer(int x,int y)
{
int p=LCA(x,y);
int lenx=d[x]-d[p],leny=d[y]-d[p];
int midlen=(lenx+leny)/2;
int nx,ny;
if(lenx>leny)
{
nx=Ancestor(x,midlen);
if((lenx+leny)%2==0)
{
printf("%d\n",nx);
}
else
{
ny=A[nx][0];
printf("%d %d\n",nx,ny);
}
}
else
{
ny=Ancestor(y,midlen);
if((lenx+leny)%2==0)
{
printf("%d\n",ny);
}
else
{
nx=A[ny][0];
printf("%d %d\n",nx,ny);
}
}
}

int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int N;
scanf("%d",&N);
BuildTree(N);
int M;
scanf("%d",&M);
while(M--)
{
int x,y;
scanf("%d%d",&x,&y);
Answer(x,y);
}
printf("\n");
}
return 0;
}