LCA(倍增法)

LCA(最近公共祖先):给你一棵树,然后给你两个点,让你找到这两个点的最近公共祖先,俩个点的公共祖先有很多,但是其中距离两个点的距离之和最小的那个祖先就是最近公共祖先。
刚学了树上倍增法求LCA
最近做了几个板子题,记录一下这种算法的基本,,,过程吧。
fa数组:倍增关键数组,fa[i][j]表示以i开始找2j步的祖先,例如fa[i][0]就表示i的20的祖先也就是i的父结点。
de数组:当前这个数的在树中的深度。
f数组:记录父节点,有的树没有给出根节点,需要用f数组找根节点,(记录入度数量也可以找到根节点)
head数组:前向星建图头数组
v数组:根结点到当前数的距离(权值)。

首先我们要预处理fa数组,用bfs遍历,更新 f a [ i ] [ j ] = f a [ f a [ i ] [ j − 1 ] ] [ j − 1 ] fa[i][j]=fa[fa[i][j-1]][j-1] fa[i][j]=fa[fa[i][j1]][j1]。同时记录深度,有要求距离的,也在预处理里面更新距离。
然后我们就可以用倍增法查找两个数a,b(de[a]>de[b],默认a是要更深,不然就swap一下)的LCA了。
在求LCA的函数中,我们要先把两个数保持在同一深度,这个过程其实用到的是二进制拆分,把差值一步步缩小,最终让深度相等,然后两个一起往上找祖先,如果两个祖先不一样就同时更新,这样a,b还是在同一深度,最终可以更新到不能一步也不能更新的地步,那么就说明再往前走一步就他俩祖先就一样了,那么这个就是他俩的最近的公共祖先。
例题一:POJ-1330 Nearest Common Ancestors
入门板题

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<algorithm>
#include<queue>
#include<vector>
#include<math.h>
//#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int N=1e4+50;
int fa[N][20];
int f[N];
int tot;
int head[N];
int de[N];
int L;
int n;
queue<int>q;
struct zxc
{
    int to,net;
}e[2*N];
void add(int u,int v)
{
    e[tot].to=v;
    e[tot].net=head[u];
    head[u]=tot++;
}
void init()
{
    memset(head,-1,sizeof(head));
    memset(f,0,sizeof(f));
    memset(fa,0,sizeof(fa));
    memset(de,0,sizeof(de));
    tot=1;
    while(!q.empty())
    {
        q.pop();
    }
}
void bfs(int x)
{
    de[x]=1;
    q.push(x);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=head[u];i!=-1;i=e[i].net)
        {
            int v=e[i].to;
            if(de[v])
            {
                continue;
            }
            de[v]=de[u]+1;
            fa[v][0]=u;
            for(int i=1;i<=L;i++)
            {
                fa[v][i]=fa[fa[v][i-1]][i-1];
            }
            q.push(v);
        }
    }
}
int lca(int a,int b)
{
    if(de[a]<de[b])
    {
        swap(a,b);
    }
    for(int i=L;i>=0;i--)
    {
        if(de[fa[a][i]]>=de[b])
        {
            a=fa[a][i];
        }
    }
    if(a==b)
    {
        return a;
    }
    for(int i=L;i>=0;i--)
    {
        if(fa[a][i]!=fa[b][i])
        {
            a=fa[a][i];
            b=fa[b][i];
        }
    }
    return fa[a][0];
}
int main()
{
    int t;
    int x,y;
    scanf("%d",&t);
    while(t--)
    {
        init();
        scanf("%d",&n);
        L=int(log(n+0.0)/log(2.0))+1;
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&x,&y);
            f[y]++;
            add(x,y);
        }
        int rot;
        for(int i=1;i<=n;i++)
        {
            if(f[i]==0)
            {
                rot=i;
                break;
            }
        }
        bfs(rot);
        scanf("%d%d",&x,&y);
        printf("%d\n",lca(x,y));
    }
    return 0;
}

例题二:POJ-1470 Closest Common Ancestors

//#include<bits/stdc++.h>
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<vector>
#include<queue>
#include<math.h>
#include<map>
using namespace std;
#define LL long long
const double PI=acos(-1.0);
const double eps=1e-8;
const int N=1e5+10;
int fa[1105][12];
//int v[1105];
int de[1105];
int f[1000];
queue<int>q;
struct zxc
{
    int to,net;
} e[10005];
int ans[1000];
int head[1005];
int tot;
int n,m,d;
double u;
char s1[5],s2[5];
void add(int u,int v)
{
    e[tot]= {v,head[u]};
    head[u]=tot++;
//    e[tot]={u,w,head[v]};
//    head[v]=tot++;
}
void bfs(int x)
{
    de[x]=1;
    //v[1]=0;
    q.push(x);
    while(!q.empty())
    {
        int x=q.front();
        q.pop();
        for(int i=head[x]; i!=-1; i=e[i].net)
        {
            int y=e[i].to;
            if(de[y])
                continue;
            de[y]=de[x]+1;
            //v[y]=v[x]+e[i].w;
            fa[y][0]=x;
            for(int i=1; i<=u; i++)
            {
                fa[y][i]=fa[fa[y][i-1]][i-1];
            }
            q.push(y);
        }
    }

}
int lca(int a,int b)
{
    if(de[a]<de[b])
    {
        swap(a,b);
    }
    for(int i=u; i>=0; i--)
    {
        if(de[fa[a][i]]>=de[b])
        {
            a=fa[a][i];
        }
    }
    if(a==b)
    {
        return a;
    }
    for(int i=u; i>=0; i--)
    {
        if(fa[a][i]!=fa[b][i])
        {
            a=fa[a][i];
            b=fa[b][i];
        }
    }
    return fa[a][0];
}
int main()
{
    while(~scanf("%d",&n))
    {
        memset(head,-1,sizeof(head));
        tot=1;
        memset(fa,0,sizeof(fa));
        // memset(v,0,sizeof(v));
        memset(de,0,sizeof(de));
        memset(ans,0,sizeof(ans));
        memset(f,0,sizeof(f));

        int x,y,z,num;
        u=log(n+0.0)/log(2.0);
        for(int i=1; i<=n; i++)
        {
            scanf("%d:(%d)",&x,&num);
            for(int i=1; i<=num; i++)
            {
                scanf("%d",&y);

                f[y]=x;
                add(x,y);
            }

        }
        while(f[y])
        {
            y=f[y];
        }
        bfs(y);
        scanf("%d",&m);
        while(m--)
        {
            scanf("%1s%d%d%1s",s1,&x,&y,s2);
            ans[lca(x,y)]++;
        }
        for(int i=1; i<=n; i++)
        {
            if(ans[i]!=0)
            {
                printf("%d:%d\n",i,ans[i]);
            }
        }
    }
    return 0;
}

例题三:HDU-4547 CD操作
注意这道题树上从上往下是可以一步到达的。

#include<bits/stdc++.h>
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<vector>
#include<queue>
#include<math.h>
#include<map>
using namespace std;
#define LL long long
const double PI=acos(-1.0);
const double eps=1e-8;
const int N=1e5+10;
int fa[N][20];
int f[N];
int de[N];
int tot;
int n,m;
int u;
char s1[45],s2[45];
map<string,int>q;
queue<int>w;
struct zxc
{
    int to,net;
} e[2*N];
int head[N];
void init()
{
    for(int i=0;i<=n;i++)
    {
        head[i]=-1;
        f[i]=0;
        de[i]=0;
    }
    memset(fa,0,sizeof(fa));
    tot=1;
}
void add(int u,int v)
{
    e[tot]= {v,head[u]};
    head[u]=tot++;
}
void bfs(int x)
{
    de[x]=1;
    w.push(x);
    while(!w.empty())
    {
        int z=w.front();
        w.pop();
        for(int i=head[z]; i!=-1; i=e[i].net)
        {
            int y=e[i].to;
            if(de[y])
            {
                continue;
            }
            de[y]=de[z]+1;
            fa[y][0]=z;
            for(int j=1; j<=u; j++)
            {
                fa[y][j]=fa[fa[y][j-1]][j-1];
            }
            w.push(y);
        }
    }
}
int lca(int x,int y)
{
    if(de[x]<de[y])
    {
        swap(x,y);
    }
    for(int i=u; i>=0; i--)
    {
        if(de[fa[x][i]]>=de[y])
        {
            x=fa[x][i];
        }
    }
    if(x==y)
    {
        return x;
    }
    for(int i=u; i>=0; i--)
    {
        if(fa[x][i]!=fa[y][i])
        {
            x=fa[x][i];
            y=fa[y][i];
        }
    }
    return fa[x][0];
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        q.clear();
        int sum=1;
        scanf("%d%d",&n,&m);
          init();
        u=int(log(n+0.0)/log(2.0))+1;
        for(int i=1; i<n; i++)
        {
            scanf("%s%s",s1,s2);
            if(!q[s1])
            {
                q[s1]=sum++;
            }
            if(!q[s2])
            {
                q[s2]=sum++;
            }
            f[q[s1]]=q[s2];
            add(q[s2],q[s1]);
        }
        int num=1;
        while(f[num])
        {
            num=f[num];
        }
        bfs(num);
        while(m--)
        {
            scanf("%s%s",s1,s2);
            int flog=1;
            int ans=lca(q[s1],q[s2]);
            if(ans==q[s2])
            {
                flog=0;
            }
            printf("%d\n",de[q[s1]]-de[ans]+flog);
        }
    }
    return 0;
}
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值