Nearest Common Ancestors
题意:求两个节点的最近公共父节点
思路:LCA tarjan
代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define N 10010
struct Edge{
int v;
int len;
int next;
}edge[N*2];
int e_tot,e_head[N],root[N];
void e_add(int a,int b,int c){
edge[e_tot].v=b;
edge[e_tot].len=c;
edge[e_tot].next=e_head[a];
e_head[a]=e_tot++;
edge[e_tot].v=a;
edge[e_tot].len=c;
edge[e_tot].next=e_head[b];
e_head[b]=e_tot++;
}
struct Query{
int v;
int id;
int next;
}que[N*2];
int q_tot,q_head[N];
void q_add(int a,int b,int c){
que[q_tot].v=b;
que[q_tot].id=c;
que[q_tot].next=q_head[a];
q_head[a]=q_tot++;
que[q_tot].v=a;
que[q_tot].id=c;
que[q_tot].next=q_head[b];
q_head[b]=q_tot++;
}
bool vis[N];
int pre[N],ans[N],n;
void init(){
e_tot=q_tot=0;
memset(vis,false,sizeof(vis));
memset(root,true,sizeof(root));
memset(e_head,-1,sizeof(e_head));
memset(q_head,-1,sizeof(q_head));
memset(ans,-1,sizeof(ans));
for(int i=1;i<=n;i++)
pre[i]=i;
}
int find(int x){
return pre[x]==x?pre[x]:pre[x]=find(pre[x]);
}
void Union(int x,int y){
x=find(x);
y=find(y);
if(x!=y){
pre[y]=x;
}
}
void Tarjan(int cur){
vis[cur]=true;
for(int i=e_head[cur];i!=-1;i=edge[i].next){
int tmp = edge[i].v;
if(!vis[tmp]){
Tarjan(tmp);
Union(cur,tmp);
}
}
for(int i=q_head[cur];i!=-1;i=que[i].next){
int tmp = que[i].v;
if(vis[tmp]){
ans[que[i].id]=find(tmp);
}
}
}
int main(){
int T,a,b,c,q,m;
scanf("%d",&T);
while(T--){
scanf("%d",&n);
init();
for(int i=0;i<n-1;i++){
scanf("%d%d",&a,&b);
e_add(a,b,0);
root[b]=false;
}
q=1;
for(int i=0;i<q;i++){
scanf("%d%d",&a,&b);
q_add(a,b,i);
}
for(int i=1;i<=n;i++){
//找到根节点后求解
if(!vis[i]&&root[i]) Tarjan(i);
}
for(int i=0;i<q;i++){
printf("%d\n",ans[i]);
}
}
return 0;
}