P3379 【模板】最近公共祖先(LCA)

原题链接

非常详细的题解见洛谷,个人见解见代码

方法一:倍增法

#include<bits/stdc++.h>
using namespace std;
#define N 500005

vector<int> G[N];//链树,以链上的元素为根节点的树
void add(int x,int y)
{
    G[x].push_back(y);
    G[y].push_back(x);
}

int fa[N][21]={0};
int depth[N]={0};
int n,m,s;

void proc()//预处理
{
    for(int k=1;k<=18;k++)//直接以最大常数算,反正也快不了多少
        for(int i=1;i<=n;i++)
            fa[i][k]=fa[fa[i][k-1]][k-1];//如果超过了,父节点就为零,即代表没有父节点
}

void build(int now,int fath)
{
    fa[now][0]=fath;
    for(int i=0;i<G[now].size();i++)
    {
        int next=G[now][i];
        if(next!=fath)//边不通来路
        {
            depth[next]=depth[now]+1;
            build(next,now);
        }
    }
}

int lca(int x,int y)
{
    if(depth[x]<depth[y])swap(x,y);

    for(int i=18;i>=0;i--)
        if(depth[fa[x][i]]>=depth[y])//最终得到的一定是相等的
            x=fa[x][i];//如果depth为零,代表该节点不存在

    //printf("%d,%d\n",x,y);
    if(x==y)return x;//如果不加这个就会返回他们的父节点,就不是最近祖先了
    //任意两个节点的父节点至少是s,所以fa[s]可以不用管
    for(int i=18;i>=0;i--)
        if(fa[x][i]!=fa[y][i])
        {
            x=fa[x][i];
            y=fa[y][i];
        }//上浮到离lca最近的点

    return fa[x][0];
}
int main()
{
    scanf("%d%d%d",&n,&m,&s);

    for(int i=1;i<=n-1;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
    }

    depth[s]=1;//要与不存在的节点分开来
    build(s,0);
    proc();

    for(int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        printf("%d\n",lca(x,y));
    }
    return 0;
}

方法二:tarjan法(但是蒟蒻只能理解成集合法

#include<bits/stdc++.h>
using namespace std;
#define N 500005
vector<int> G[N];
vector< pair<int,int> > Q[N];
int ans[N]={0};
int vis[N]={0};
int fa[N]={0};

int finds(int now)
{
    return now==fa[now]?now:fa[now]=finds(fa[now]);
}

void ss(int now)
{
    fa[now]=now;
    vis[now]=1;
    for(int i=0;i<G[now].size();i++)
    {
        int next=G[now][i];
        if(!vis[next])
        {
            ss(next);//lca有两种,一种在一条直链上,另一种不在同一条链上,对于不在同一条链上的情况的解释:ss内出现的且第一次出现的询问都归结到该点
            fa[next]=now;
        }
    }
    for(int i=0;i<Q[now].size();i++)
    {
        int ano=Q[now][i].first;
        if(vis[ano])
        {
            ans[Q[now][i].second]=finds(ano);
        }
    }
}



void addedge(int x,int y)
{
    G[x].push_back(y);
    G[y].push_back(x);
}
void addques(int x,int y,int id)
{
    Q[x].push_back(make_pair(y,id));
    Q[y].push_back(make_pair(x,id));
}
int main()
{
    int n,m,s;
    scanf("%d%d%d",&n,&m,&s);

    for(int i=1;i<=n-1;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        addedge(x,y);
    }

    for(int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        addques(x,y,i);
    }

    ss(s);

    for(int i=1;i<=m;i++)
    {
        printf("%d\n",ans[i]);
    }
    return 0;
}


posted @   纯粹的  阅读(27)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· Manus的开源复刻OpenManus初探
· AI 智能体引爆开源社区「GitHub 热点速览」
· 从HTTP原因短语缺失研究HTTP/2和HTTP/3的设计差异
· 三行代码完成国际化适配,妙~啊~
点击右上角即可分享
微信分享提示