#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#define MAXN 1010001
using namespace std;
int n,m,to[MAXN],nxt[MAXN],s,head[MAXN],e,pos[MAXN],dfn[MAXN],u,v,dep[MAXN],lg[MAXN];
int st[MAXN][20];
inline void add( int x,int y)
{
e++;
to[e]=y;
nxt[e]=head[x];
head[x]=e;
}
void dfs(int x,int f)
{
dfn[++e]=x;
pos[x]=e;
for(int i=head[x];i;i=nxt[i])
{
int y=to[i];
if(y!=f)
{
dep[y]=dep[x]+1;
dfs(y,x);
dfn[++e]=x;
}
}
}
int lca(int x,int y)
{
int pc,pcc,l;
if(pos[x]>pos[y]) swap(x,y);
pc=pos[x];
pcc=pos[y];
l=lg[pcc-pc+1];
if(dep[st[pc][l]]<dep[st[pcc-(1<<l)+1][l]])
return st[pc][l];
else
return st[pcc-(1<<l)+1][l];
}
int main()
{
// freopen("1.txt","r",stdin);
cin>>n>>m>>s;
memset(dep,0x7f,sizeof(dep));
for(int i=1;i<n;i++)
{
scanf("%d%d",&u,&v);
add(u,v);
add(v,u);
}
e=0;
dep[s]=0;
dfs(s,0);
lg[0]=-1;
for(int i=1;i<=e;i++)
{
st[i][0]=dfn[i];
}
//i<=n就行了
for(int i=1;i<=e;i++)
{
if((i&(i-1))==0) lg[i]=lg[i-1]+1;
else lg[i]=lg[i-1];
}
for(int i=1;i<=lg[e];i++) //j+(1<<i)-1<=e就行了
{
for(int j=1;j+(1<<i)-1<=e;j++)
{
//右边的dep[st[j+(1<<(i-1)][i-1]]!!!
// ↑
//下面else里也是
if(dep[st[j][i-1]]<dep[st[j+(1<<(i-1))][i-1]])
st[j][i]=st[j][i-1];
else st[j][i]=st[j+(1<<(i-1))][i-1];
}
}
for(int i=1;i<=m;i++)
{
scanf("%d%d",&u,&v);
printf("%d\n",lca(u,v));
}
}
RMQ求LCA
最新推荐文章于 2021-06-04 11:03:05 发布