BZOJ1602: [Usaco2008 Oct]牧场行走

1602: [Usaco2008 Oct]牧场行走

Time Limit: 5 Sec  Memory Limit: 64 MB
Submit: 1084  Solved: 556
[Submit][Status]

Description

N头牛（2<=n<=1000）别人被标记为1到n，在同样被标记1到n的n块土地上吃草，第i头牛在第i块牧场吃草。 这n块土地被n-1条边连接。 奶牛可以在边上行走，第i条边连接第Ai，Bi块牧场，第i条边的长度是Li（1<=Li<=10000）。 这些边被安排成任意两头奶牛都可以通过这些边到达的情况，所以说这是一棵树。 这些奶牛是非常喜欢交际的，经常会去互相访问,他们想让你去帮助他们计算Q(1<=q<=1000)对奶牛之间的距离。

Input

*第一行：两个被空格隔开的整数：N和Q

 *第二行到第n行：第i+1行有两个被空格隔开的整数：AI，BI，LI

*第n+1行到n+Q行：每一行有两个空格隔开的整数：P1，P2，表示两头奶牛的编号。

Output

*第1行到第Q行：每行输出一个数，表示那两头奶牛之间的距离。

Sample Input

4 2
2 1 2
4 3 2
1 4 3
1 2
3 2

Sample Output

2
7

HINT

 

Source

资格赛

题解：

不想写倍增了，把树链剖分的代码贴上改了改就A了。。。

每个节点保存的是到父亲的距离，所以对LCA要特判

代码：

const maxn=1000+100;
type node1=record
     go,next,z:longint;
     end;
     node2=record
     l,r,mid,sum:longint;
     end;

var  e:array[0..2*maxn] of node1;
     t:array[0..4*maxn] of node2;
     p,a,v,fa,s,head,dep,son,top:array[0..maxn] of longint;
     i,n,m,x,y,z,sz,ans,tot:longint;
     ch:char;
     procedure swap(var x,y:longint);
      var t:longint;
      begin
        t:=x;x:=y;y:=t;
      end;
     procedure insert(x,y,z:longint);
      begin
        inc(tot);
        e[tot].go:=y;e[tot].z:=z;e[tot].next:=head[x];head[x]:=tot;
      end;
     function min(x,y:longint):longint;
      begin
        if x<y then exit(x) else exit(y);
      end;
     function max(x,y:longint):longint;
      begin
        if x>y then exit(x) else exit(y);
      end;

procedure dfs1(x:longint);
 var i,j,y:longint;
 begin
   j:=0;s[x]:=1;
   i:=head[x];
   while i<>0 do
    begin
     y:=e[i].go;
     if dep[y]=0 then
      begin
        dep[y]:=dep[x]+1;
        fa[y]:=x;v[y]:=e[i].z;
        dfs1(y);
        inc(s[x],s[y]);
        if s[y]>s[j] then j:=y;
      end;
     i:=e[i].next;
    end;
   son[x]:=j;
 end;
procedure dfs2(x,chain:longint);
 var i,y:longint;
 begin
   inc(sz);p[x]:=sz;top[x]:=chain;
   if son[x]<>0 then dfs2(son[x],chain);
   i:=head[x];
   while i<>0 do
     begin
      y:=e[i].go;
      if (p[y]=0) and (y<>son[x]) then dfs2(y,y);
      i:=e[i].next;
     end;
 end;
procedure pushup(k:longint);
 begin
    t[k].sum:=t[k<<1].sum+t[k<<1+1].sum;
 end;

procedure build(k,x,y:longint);
 begin
   with t[k] do
    begin
     l:=x;r:=y;mid:=(l+r)>>1;
     if l=r then begin sum:=a[l];exit;end;
     build(k<<1,l,mid);build(k<<1+1,mid+1,r);
     pushup(k);
    end;
 end;
function getsum(k,x,y:longint):longint;
 begin
   with t[k] do
    begin
     if (l=x) and (r=y) then exit(sum);
     if y<=mid then exit(getsum(k<<1,x,y))
     else if x>mid then exit(getsum(k<<1+1,x,y))
     else exit(getsum(k<<1,x,mid)+getsum(k<<1+1,mid+1,y));
    end;
 end;
procedure init;
 begin
   readln(n,m);
   for i:=1 to n-1 do begin readln(x,y,z);insert(x,y,z);insert(y,x,z);end;
   dep[1]:=1;
   dfs1(1);
   dfs2(1,1);
   for i:=2 to n do a[p[i]]:=v[i];
   build(1,1,n);
 end;
procedure solvesum;
 begin
   ans:=0;
   readln(x,y);
     while top[x]<>top[y] do
     begin
      if dep[top[x]]<dep[top[y]] then swap(x,y);
      inc(ans,getsum(1,p[top[x]],p[x]));
      x:=fa[top[x]];
     end;
   if dep[x]>dep[y] then swap(x,y);
   inc(ans,getsum(1,p[x],p[y]));
   if p[x]<>1 then dec(ans,v[x]);
   writeln(ans);
 end;

procedure main;
 begin
   for i:=1 to m do solvesum;
 end;

begin
  assign(input,'input.txt');assign(output,'output.txt');
  reset(input);rewrite(output);
  init;
  main;
  close(input);close(output);
end.