给一棵树,查询两个点的最近公共祖先。还是模板题,不过数据给的时候是按顺序给的,所以要直接存成从根节点到叶子节点的有向图,同时存一下每个点的入度用来找根节点,然后从根节点开始做一遍tarjan就行..
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <memory.h>
#include <cmath>
using namespace std;
typedef long long ll;
const int maxn=40000+40;
struct node
{
int w,v;
int next;
}edge[maxn<<1],query[500];
bool vis[maxn];
int g[maxn],gg[maxn];
int res[maxn][3];
int m,n,k,p,q;
int en,eq;
int test;
int fa[maxn];
int dis[maxn];
int find(int x)
{
if (x!=fa[x]) return fa[x]=find(fa[x]);
return x;
}
void tarjan(int u)
{
vis[u]=true;
fa[u]=u;
for (int j=gg[u]; j!=-1; j=query[j].next)
{
if (vis[query[j].v])
{
res[query[j].w][2]=find(query[j].v);
}
}
for (int j=g[u]; j!=-1; j=edge[j].next)
{
if (!vis[edge[j].v])
{
dis[edge[j].v]=dis[u]+edge[j].w;
tarjan(edge[j].v);
fa[edge[j].v]=u;
}
}
}
int ideg[maxn];
int main()
{
// freopen("in.txt","r",stdin);
scanf("%d",&test);
while (test--)
{
memset(vis,false,sizeof vis);
memset(dis,0,sizeof dis);
memset(ideg,0,sizeof ideg);
scanf("%d",&n);
// scanf("%d",&m);
m=1;
memset(g,-1,sizeof g);
memset(gg,-1,sizeof gg);
en=eq=0;
int x,y,z;
int root=-1;
for (int i=1; i<n; i++)
{
scanf("%d%d",&x,&y);
ideg[y]++;
if (root==-1) root=x;
z=1;
edge[en].w=z;
edge[en].v=y;
edge[en].next=g[x];
g[x]=en;
en++;
}
for (int i=1; i<=m; i++)
{
scanf("%d%d",&x,&y);
query[eq].v=y;
query[eq].w=i;
query[eq].next=gg[x];
gg[x]=eq;
eq++;
query[eq].v=x;
query[eq].w=i;
query[eq].next=gg[y];
gg[y]=eq;
eq++;
res[i][0]=x;
res[i][1]=y;
}
for (int i=1; i<=n; i++)
if (ideg[i]==0)
{
root=i;
break;
}
dis[root]=0;
tarjan(root);
cout<<res[1][2]<<endl;
}
return 0;
}