题目链接
题意:给你一棵树,后面是前面的父节点,k次查询,如果a是b的祖先输出1,b是a的祖先输出2,否则输出0;
思路:根据最近公共祖先判断得出结果;
/*LCA(离线算法)
主函数除建边外还应调用
init();
dir[1]=0;
tarjan(1);
*/
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn=1e5+10;
struct zp
{
int a,b;
}node[maxn*4];
struct note
{
int u,v,w,lca,next;
}edge[maxn*4],edge1[maxn*4];
int head[maxn*4],ip,head1[maxn*4],ip1;///需要建两次边。1,该树的边 2,需要查询的两点
int m,n;
int father[maxn*4],vis[maxn*4],dir[maxn*4];
///依次表示u点的祖先、标记是否访问过,到根节点的距离
void init()
{
memset(vis,0,sizeof(vis));
memset(dir,0,sizeof(dir));
memset(head,-1,sizeof(head));
memset(head1,-1,sizeof(head1));
ip=ip1=0;
}
void addedge(int u,int v,int w)
{
edge[ip].v=v,edge[ip].w=w,edge[ip].next=head[u],head[u]=ip++;
}
void addedge1(int u,int v)
{
edge1[ip1].u=u,edge1[ip1].v=v,edge1[ip1].lca=-1,edge1[ip1].next=head1[u],head1[u]=ip1++;
}
int Find(int x)
{
if(father[x]==x)
return x;
return father[x]=Find(father[x]);
}
void Union(int x,int y)
{
x=Find(x);
y=Find(y);
if(x!=y)
father[y]=x;
}
void tarjan(int u)
{
vis[u]=1;
father[u]=u;
for(int i=head[u];i!=-1;i=edge[i].next)
{
int v=edge[i].v;
int w=edge[i].w;
if(!vis[v])
{
dir[v]=dir[u]+w;
tarjan(v);
Union(u,v);
}
}
for(int i=head1[u];i!=-1;i=edge1[i].next)
{
int v=edge1[i].v;
if(vis[v])
{
edge1[i].lca=edge1[i^1].lca=father[Find(v)];
}
}
}
int main()
{
int n;
while(~scanf("%d",&n))
{
init();
int root;
for(int i=0;i<n;i++)//建树
{
int a,b;
scanf("%d%d",&a,&b);
if(b==-1)
root=a;
else
addedge(b,a,1);
}
int k;
scanf("%d",&k);
for(int i=0;i<k;i++)//离线查询建边
{
int a,b;
scanf("%d%d",&a,&b);
node[i].a=a;
node[i].b=b;
addedge1(a,b);
addedge1(b,a);
}
dir[root]=0;
tarjan(root);
for(int i=0;i<k*2;i+=2)
{
int a=node[i/2].a,b=node[i/2].b;
int tmp=edge1[i].lca;//a和b的公共最近公共祖先
if(tmp==a)
printf("1\n");
else if(tmp==b)
printf("2\n");
else
printf("0\n");
}
}
return 0;
}