祖孙询问 纪中3054 LCA

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.