Description
给出N,Q
代表一个树有N个点 ,树的根为1
Q代表有Q个询问,询问A,B的最近公共祖先是哪一个.
代表一个树有N个点 ,树的根为1
Q代表有Q个询问,询问A,B的最近公共祖先是哪一个.
Input
第一行给出N,Q
下面N-1行描述这个树,格式为A,B,代表A为B的Son
现来下Q行
给出A,B,代表进行询问A,B的LCA是哪一个.
下面N-1行描述这个树,格式为A,B,代表A为B的Son
现来下Q行
给出A,B,代表进行询问A,B的LCA是哪一个.
Output
针对每个询问输出结果
Sample Input
4 2
2 1
3 2
4 2
3 4
4 2
Sample Output
2
2
题解:一道裸的求最近公共祖先(LCA)的模板题,用来练手或入门的。
直接附上代码:
#include<bits/stdc++.h>
const int maxn=300001;
using namespace std;
struct Edge{int v,next,z;} edge[maxn*2];
int cnt,deep[maxn],fa[maxn][25],n,m,head[maxn],ans;
void add(int u,int v)
{
edge[++cnt].v=v;
edge[cnt].next=head[u];
head[u]=cnt;
}
int read()
{
int x=0,f=1;
char ch=getchar();
while(ch>'9'||ch<'0'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
char digit[10];
void write(int x){
if (!x){putchar('0');return;}
int len=0;
while (x)digit[len++]=x%10,x/=10;
for (int i=len-1;~i;i--)putchar(digit[i]+'0');
}
void dfs(int now,int father,int dep)
{
fa[now][0]=father;
deep[now]=dep;
for(int i=head[now];i;i=edge[i].next)
if(father!=edge[i].v)dfs(edge[i].v,now,dep+1);
}
void getf()
{
for(int j=1;j<=20;j++)
for(int i=1;i<=n;i++)
fa[i][j]=fa[fa[i][j-1]][j-1];
}
int gets(int a,int b)
{
for(int i=0;i<=20;i++)
if(b&(1<<i))a=fa[a][i];
return a;
}
int lca(int u,int v)
{
if(deep[u]<deep[v])swap(u,v);
u=gets(u,deep[u]-deep[v]);
if(u==v)return u;
for(int i=20;i>=0;i--)
if(fa[u][i]!=fa[v][i])
u=fa[u][i],v=fa[v][i];
return fa[u][0];
}
int main()
{
n=read();m=read();
for(int i=1,x,y;i<n;i++)
x=read(),y=read(),add(x,y),add(y,x);
dfs(1,1,0);
getf();
for(int i=1,u,v;i<=m;i++)
{
u=read(),v=read();
int l=lca(u,v);
write(l);
printf("\n");
}
}