洛谷3379 最近公共祖先

题目描述

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

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

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

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

输出格式：

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

dfs+并查集离线O(n+q)求LCA。
对于每个点，先把并查集的父亲标记成自己，dfs完一棵子树之后把子树加入自己的并查集中。然后考虑所有与自己有关的询问，如果另一个点已经被访问而且这一询问尚没有得出答案，那么答案就是另一个点的父亲。

#include<cstdio>#include<cstring>int e_fir[500010],e_ne[1000010],e_to[1000010],e_tot,q_fir[500010],q_ne[1000010],q_to[1000010],q_num[1000010],q_tot,ans[500010],fa[500010],n,m,r;int rd(){    int x=0;    char c=getchar();    while (c<'0'||c>'9') c=getchar();    while (c>='0'&&c<='9')    {        x=x*10+c-'0';        c=getchar();    }    return x;}int find(int x){    return x==fa[x]?x:fa[x]=find(fa[x]);}void add_edge(int f,int t){    e_tot++;    e_ne[e_tot]=e_fir[f];    e_fir[f]=e_tot;    e_to[e_tot]=t;}void add_qry(int f,int t,int num){    q_tot++;    q_ne[q_tot]=q_fir[f];    q_fir[f]=q_tot;    q_to[q_tot]=t;    q_num[q_tot]=num;}void dfs(int u,int f){    int i,v,x,y,z;    fa[u]=u;    for (i=e_fir[u];i;i=e_ne[i])      if ((v=e_to[i])!=f)      {        dfs(v,u);        fa[find(v)]=find(u);      }    for (i=q_fir[u];i;i=q_ne[i])      if (find(q_to[i])&&!ans[q_num[i]])        ans[q_num[i]]=fa[q_to[i]];}int main(){    int i,j,k,p,q,x,y,z;    scanf("%d%d%d",&n,&m,&r);    for (i=1;i<n;i++)    {        x=rd();        y=rd();        add_edge(x,y);        add_edge(y,x);    }    for (i=1;i<=m;i++)    {        x=rd();        y=rd();        add_qry(x,y,i);        add_qry(y,x,i);    }    dfs(r,-1);    for (i=1;i<=m;i++)      printf("%d\n",ans[i]);}