题目链接:http://poj.org/problem?id=1330
题目大意:给出n个父子关系,不会出现回路,求最原始祖先(根节点)。
照大佬的板子自己手码了一遍,还挺长,,,
AC代码:
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
#include<limits>
#include<vector>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
const int maxx=10010;
int F[maxx];
int r[maxx];
bool vis[maxx];
int anc[maxx];
struct node
{
int to,nxt;
}edge[2*maxx];
int head[maxx];
int tol;
void addedge(int a,int b)
{
edge[tol].to=b;
edge[tol].nxt=head[a];
head[a]=tol++;
edge[tol].to=a;
edge[tol].nxt=head[b];
head[b]=tol++;
}
struct Query
{
int q;
int nxt;
int index;
}query[2*maxx];
int ans[2*maxx];
int cnt;
int h[maxx];
int tt;
int Q;
void add_query(int a,int b,int i)
{
query[tt].q=b;
query[tt].nxt=h[a];
query[tt].index=i;
h[a]=tt++;
query[tt].q=a;
query[tt].nxt=h[b];
query[tt].index=i;
h[b]=tt++;
}
void init(int n)
{
for(int i=1;i<=n;i++)
{
F[i]=-1;
r[i]=1;
vis[i]=0;
anc[i]=0;
tol=0;
tt=0;
cnt=0;
}
memset(head,-1,sizeof(head));
memset(h,-1,sizeof(h));
}
int findd(int x)
{
if(F[x]==-1)
return x;
return F[x]=findd(F[x]);
}
void Union(int x,int y)
{
int t1=findd(x);
int t2=findd(y);
if(t1!=t2)
{
if(r[t1]<=r[t2])
{
F[t1]=t2;
r[t2]+=r[t1];
}
else
{
F[t2]=t1;
r[t1]+=r[t2];
}
}
}
void LCA(int u)
{
anc[u]=u;
vis[u]=1;
for(int i=head[u];i!=-1;i=edge[i].nxt)
{
int v=edge[i].to;
if(vis[v])
continue;
LCA(v);
Union(u,v);
anc[findd(u)]=u;
}
for(int i=h[u];i!=-1;i=query[i].nxt)
{
int v=query[i].q;
if(vis[v])
{
ans[query[i].index]=anc[findd(v)];
cnt++;
}
}
}
bool flag[maxx];
int main()
{
int t;
int u,v;
int N;
scanf("%d",&t);
while(t--)
{
scanf("%d",&N);
init(N);
memset(flag,0,sizeof(flag));
for(int i=1;i<N;i++)
{
scanf("%d%d",&u,&v);
flag[v]=1;
addedge(u,v);
}
Q=1;
scanf("%d%d",&u,&v);
add_query(u,v,0);
int root;
for(int i=1;i<=N;i++)
{
if(!flag[i])
{
root=i;
break;
}
}
LCA(root);
for(int i=0;i<Q;i++)
{
printf("%d",ans[i]);
}
printf("\n");
}
return 0;
}
我一定可以的!!!