tarjan求lca的神奇

佚名 5年前 (2019-04-21) 随笔 1984人围观抢沙发百度已收录

题目描述

如题，给定一棵有根多叉树，请求出指定两个点直接最近的公共祖先。

输入输出格式

输入格式：

SRE实战互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。

第一行包含三个正整数N、M、S，分别表示树的结点个数、询问的个数和树根结点的序号。

接下来N-1行每行包含两个正整数x、y，表示x结点和y结点之间有一条直接连接的边（数据保证可以构成树）。

接下来M行每行包含两个正整数a、b，表示询问a结点和b结点的最近公共祖先。

输出格式：

输出包含M行，每行包含一个正整数，依次为每一个询问的结果。

输入输出样例

输入样例#1：

输出样例#1：

说明

时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=10，M<=10

对于70%的数据：N<=10000，M<=10000

对于100%的数据：N<=500000，M<=500000

~~~~~~~~~~~~~~~华丽的分割线~~~~~~~~~~~~~~~

对于此题，常规解法就是暴力。

↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑

炸死你。

倍增，可以有效降低复杂度，我也是学了两次之后掌握了它。

这里，我想记录下不久前学习的tarjan求lca

（tarjan真是好东西）

说它是tarjan，其实给我的感觉就是套了一个tarjan的名字罢了。（类似我的dijksPA）其实它就是一个dfs的神奇过程。

先放代码再说

#include<bits/stdc++.h>
using namespace std;
const int maxn=500005;
int n,m,p;
struct edge
{
    int next,to,lca;
}e[maxn<<1],e1[maxn<<1];
int head[maxn<<1],cnt,head1[maxn<<1];
inline int addedge(int from,int to)//前向星存图
{
    e[++cnt].next=head[from];
    e[cnt].to=to;
    head[from]=cnt;
}
inline int add (int from,int to)//前向星存储查询
{
    e1[++cnt].next=head1[from];
    e1[cnt].to=to;
    head1[from]=cnt;
}
int f[maxn<<1],vis[maxn<<1];
int find(int x)//并查集找爹函数
{
    if(f[x]!=x)
    f[x]=find(f[x]);
    return f[x];
}
int tarjan(int u)//神奇的dfs
{
    f[u]=u;//爸爸
    vis[u]=1;//走过
    for(int i=head[u];i;i=e[i].next)
    {
        int v=e[i].to;
        if(!vis[v])
        {
            tarjan(v);
            f[v]=u;//爹
        }
    }
    for(int i=head1[u];i;i=e1[i].next)//询问的图
    {
        int v=e1[i].to;
        if(vis[v])
        {
            e1[i].lca=find(v);
            if(i%2==1)
            e1[i+1].lca=e1[i].lca;//
            else
            e1[i-1].lca=e1[i].lca;
        }
    }
}


int main()
{
    scanf("%d%d%d",&n,&m,&p);
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        addedge(x,y);
        addedge(y,x);
    }
    for(int i=1;i<=n;i++)
    f[i]=i;//初始化并查集
    cnt=0;//因为上面村边用过cnt，所以把它拍成0
    for(int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    tarjan(p);//从根节点开始dfs
    for(int i=1;i<=m;i++)
    {
        printf("%d\n",e1[i*2].lca);
    }
    return 0;
}