最近公共祖先（LCA） 洛谷 3379 LCA

题目描述

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

分析

直接上tarjan
水了一道模板（c++的）。
顺便复习一下。

code

#include<iostream>#include<cstring>#include<cstdio>#include<cmath>#include<algorithm>using namespace std;struct arr{    int x,y;    int w;    int next;}edge[2000000],problem[2000000];//edge存边，problem存问题。 int ls[2000000],ls_p[2000000];//对应上面。 int n,m,root;int nn,nm;int v[2000000];//记录是否来过。 int f[2000000];//并查集数组。 int find(int x)//并查集查找。 {    if (f[x]==x) return x;    else{        f[x]=find(f[x]);        return f[x];    }}void merge(int x,int y)//并查集合并。{    int x1,y1;    x1=find(x);    y1=find(y);    f[x1]=y1;}void add(int x,int y)//添加边，建树。 {    nn++;    edge[nn].x=x;    edge[nn].y=y;    edge[nn].next=ls[x];    ls[x]=nn;    nn++;    edge[nn].x=y;    edge[nn].y=x;    edge[nn].next=ls[y];    ls[y]=nn;}void add_p(int x,int y)//tarjan算法，要求储存问题。 {    nm++;    problem[nm].x=x;    problem[nm].y=y;    problem[nm].next=ls_p[x];    ls_p[x]=nm;    nm++;    problem[nm].x=y;    problem[nm].y=x;    problem[nm].next=ls_p[y];    ls_p[y]=nm;}void tarjan(int r,int x)//tarjan算法正文。 {    for (int i=ls[x];i!=0;i=edge[i].next)    {        if (edge[i].y==r) continue;        tarjan(x,edge[i].y);        v[edge[i].y]=1;        merge(edge[i].y,x);    }    for (int i=ls_p[x];i!=0;i=problem[i].next)    {        if (v[problem[i].y])                problem[i].w=find(problem[i].y);    }}int main(){    scanf("%d%d%d",&n,&m,&root);    for (int i=1;i<=n-1;i++)    {        int x,y;        scanf("%d %d",&x,&y);        add(x,y);    }    for (int i=1;i<=m;i++)    {        int x,y;        scanf("%d %d",&x,&y);        add_p(x,y);    }    // 输入。     for (int i=1;i<=n;i++)        f[i]=i;    //并查集初始化。     tarjan(0,root);    for (int i=1;i<=nm;i++)    {        if (!problem[i].w) continue;        printf("%d\n",problem[i].w);    }}