[模板]最近公共祖先LCA

Description

给出N,Q 
代表一个树有N个点 ,树的根为1
Q代表有Q个询问,询问A,B的最近公共祖先是哪一个.

Input

第一行给出N,Q 
下面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");
    }
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值