【RMQ&LCA】[Usaco2010 Hol]cowpol…

[Usaco2010 Hol]cowpol 奶牛政坛

Time Limit:10000MS  Memory Limit:65536K
Total Submit:11 Accepted:5
Case Time Limit:1000MS

Description

农夫约翰的奶牛住在N (2 <= N <= 200,000)片不同的草地上，标号为1到N。恰好有N-1条单位长度的双向道路，用各种各样的方法连接这些草地。而且从每片草地出发都可以抵达其他所有草地。也就是说，这些草地和道路构成了一种叫做树的图。输入包含一个详细的草地的集合，详细说明了每个草地的父节点P_i (0 <= P_i <= N)。根节点的P_i == 0, 表示它没有父节点。因为奶牛建立了1到K一共K (1 <= K <= N/2)个政党。每只奶牛都要加入某一个政党，其中，
第i只奶牛属于第A_i (1 <= A_i <= K)个政党。而且每个政党至少有两只奶牛。
这些政党互相吵闹争。每个政党都想知道自己的“范围”有多大。其中，定义一个政党的范围是这个政党离得最远的两只奶牛（沿着双向道路行走）的距离。
比如说，记为政党1包含奶牛1，3和6，政党2包含奶牛2，4和5。这些草地的连接方式如下图所
示（政党1由-n-表示）：


政党1最大的两只奶牛的距离是3（也就是奶牛3和奶牛6的距离）。政党2最大的两只奶牛的距离是2（也就是奶牛2和4，4和5，还有5和2之间的距离）。
帮助奶牛们求出每个政党的范围。

Input

* 第一行: 两个由空格隔开的整数: N 和 K

* 第2到第N+1行: 第i+1行包含两个由空格隔开的整数: A_i和P_i

Output

* 第1到第K行: 第i行包含一个单独的整数，表示第i个政党的范围。

Sample Input

6 2
1 3
2 1
1 0
2 1
2 1
1 5

Sample Output

3
2

Source

Gold

 

LCA问题。

对于求一个政党中的最大距离，一开始是一个一个枚举，结果肯定TLE。

然后看大神做法：先找到每个党在树中最深的节点。它一定是最长的路的一端。

 

type

 arrt=record

 x,y,next:longint;

 end;

 arrq=record

 a,next:longint;

 end;

vartype

 arrt=record

 x,y,next:longint;

 end;

 arrq=record

 a,next:longint;

 end;

var

 m,k,father:longint;

 par:array[1..200000+1]of arrq;

 lastp:array[1..100000+1]of longint;

 tree:array[1..200000+1]of arrt;

 last:array[0..200000+1]of longint;

 d:array[1..200000+1]of longint;

 mark:array[0..200000+1]of boolean;

 f:array[1..200000+1]of longint;

 max:array[0..200000+1]of longint;

 maxd,maxdi:array[1..200000+1]of longint;

 

procedure dfs(x:longint);

var

 i,y,k:longint;

begin

 y:=last[x];

 while y<>0 do

  begin

   k:=tree[y].y;

   if not mark[k] then

    begin

    d[k]:=d[x]+1;

    mark[k]:=true;

    dfs(k);

    end;

   y:=tree[y].next;

  end;

end;

 

function getfather(x:longint):longint;

begin

 if f[x]=x then exit(x)

 else

  begin

  f[x]:=getfather(f[x]);

  exit(f[x]);

  end;

end;

 

procedure lca(x,s:longint);

var

 i,y,k,p,dd:longint;

begin

 f[x]:=x;

 y:=last[x];

 while y<>0 do

   begin

    k:=tree[y].y;

    if k<>s then

     begin

     lca(k,x);

     f[k]:=x;

     end;

    y:=tree[y].next;

   end;

 mark[x]:=true;

 p:=par[x].a;

 

 if x<>maxdi[p] then

   begin

   y:=maxdi[p];

   if not mark[y] then exit;

    dd:=d[x]+d[y]-d[getfather(y)]*2;

      if dd>max[p] then

         max[p]:=dd;

   end

 else

 begin

 y:=lastp[p];

 while y<>0 do

   begin

   if x<>y then

    if mark[y] then

      begin

      dd:=d[x]+d[y]-d[getfather(y)]*2;

      if dd>max[p] then

         max[p]:=dd;

      end;

    y:=par[y].next;

   end;

  end;

end;

 

procedure init;

var

 i,a,p:longint;

begin

 read(m,k);

 for i:=1 to m do

  begin

  readln(a,p);

  if p<>0 then

  begin

    tree[i].x:=p;

    tree[i].y:=i;

    tree[i].next:=last[p];

    last[p]:=i;

  end

  else

  father:=i;

 

  par[i].a:=a;

  par[i].next:=lastp[a];

  lastp[a]:=i;

  end;

end;

 

procedure print;

var

 i:longint;

begin

 for i:=1 to k do writeln(max[i]);

end;

 

procedure max_d;

var

 y,i:longint;

begin

 for i:=1 to k do

  begin

  y:=lastp[i];

  while y<>0 do

   begin

   if d[y]>maxd[i] then begin maxd[i]:=d[y]; maxdi[i]:=y; end;

   y:=par[y].next;

   end;

  end;

end;

 

begin

 init;

 fillchar(d,sizeof(d),0);

 fillchar(mark,sizeof(mark),0);

 mark[father]:=true;

 dfs(father);

 fillchar(mark,sizeof(mark),0);

 max_d;

 lca(father,0);

 print;

end.