Description
已知一棵n个节点的有根树。有m个询问。每个询问给出了一对节点的编号x和y,询问x与y的祖孙关系。
Input
输入第一行包括一个整数n表示节点个数。
接下来n行每行一对整数对a和b表示a和b之间有连边。如果b是-1,那么a就是树的根。
第n+2行是一个整数m表示询问个数。
接下来m行,每行两个正整数x和y。
Output
对于每一个询问,输出1:如果x是y的祖先,输出2:如果y是x的祖先,否则输出0。
分析
水的要死的LCA
建树的时候要注意
代码
type
arr=record
w:longint;
x,y:longint;
flag:boolean;
next:longint;
end;
var
a,g:array[1..100010] of arr;
fa,v:array[1..100010] of longint;
ls,ls1:array[1..100010] of longint;
nm,nm1:longint;
n,m:longint;
i,j,k:longint;
root:longint;
procedure add(x,y:longint);
begin
nm:=nm+1;
a[nm].x:=x;
a[nm].y:=y;
a[nm].next:=ls[x];
ls[x]:=nm;
end;
function find(x:longint):longint;
begin
if fa[x]=x
then exit(x)
else
begin
find:=find(fa[x]);
fa[x]:=find;
end;
end;
procedure union(x,y:longint);
var
i,j:longint;
begin
i:=find(x); j:=find(y);
fa[i]:=j;
end;
procedure dfs(r:longint);
var
i,j,k:longint;
begin
i:=ls[r];
while i<>0 do
begin
if a[i].flag
then begin
i:=a[i].next;
continue;
end;
dfs(a[i].y);
union(a[i].y,a[i].x);
v[a[i].y]:=1;
i:=a[i].next;
end;
i:=ls1[r];
while i<>0 do
begin
if v[g[i].y]=1
then begin
if find(g[i].y)=r
then
if g[i].flag
then g[i].w:=1
else g[i].w:=2;
end;
i:=g[i].next;
end;
end;
procedure main(r,x:longint);
var
i,j,k:longint;
begin
i:=ls[x];
while i<>0 do
begin
if a[i].y=r
then begin
a[i].flag:=true;
i:=a[i].next;
continue;
end;
main(x,a[i].y);
i:=a[i].next;
end;
end;
begin
fillchar(fa,sizeof(fa),0);
fillchar(ls,sizeof(ls),0);
fillchar(ls1,sizeof(ls1),0);
fillchar(v,sizeof(v),0);
nm:=0; nm1:=0;
readln(n);
for i:=1 to n do
begin
readln(j,k);
if k=-1
then begin
root:=j;
continue;
end;
add(j,k);
add(k,j);
end;
readln(m);
for i:=1 to m do
begin
readln(j,k);
nm1:=nm1+1;
g[nm1].x:=j;
g[nm1].y:=k;
g[nm1].next:=ls1[j];
g[nm1].flag:=true;
ls1[j]:=nm1;
nm1:=nm1+1;
g[nm1].x:=k;
g[nm1].y:=j;
g[nm1].next:=ls1[k];
ls1[k]:=nm1;
end;
main(0,root);
for i:=1 to 50000 do
fa[i]:=i;
dfs(root);
i:=1;
while i<=nm1 do
begin
if (g[i].w=0) and (g[i+1].w=0)
then writeln(0);
if (g[i].w=0) and (g[i+1].w<>0)
then writeln(g[i+1].w);
if (g[i].w<>0) and (g[i+1].w=0)
then writeln(g[i].w);
i:=i+2;
end;
end.