题目描述
如题,给定一棵有根多叉树,请求出指定两个点直接最近的公共祖先。
分析
直接上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);
}
}