[BZOJ602][Usaco2008 Oct]牧场行走

[Usaco2008 Oct]牧场行走

Time Limit: 5 Sec Memory Limit: 64 MB

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

Source

资格赛

题解

• 对每一组a,b进行一次SPFA
• 复杂度 O(N2)

var
 s,dist:array[0..1000]of longint;
 w:array[0..3000,1..3]of longint;
 t:array[0..10000]of longint;
 i,j,k:longint;
 n,q:longint;
 a,b,c,len,head,tail,v,tt:longint;
procedure init(a,b,c:longint);
begin
 w[len,1]:=b; w[len,2]:=c;
 if w[a,3]=0
 then w[a,3]:=len else w[w[a,1],3]:=len;
 w[a,1]:=len; inc(len);
end;

begin
 readln(n,q); len:=n+1;
 for i:=1 to n-1 do
  begin
   readln(a,b,c);
   init(a,b,c); init(b,a,c);
  end;
 for i:=1 to q do
  begin
   readln(a,b);
   for j:=1 to n do
    begin dist[j]:=100000000; s[j]:=0; end;
   dist[a]:=0; s[a]:=1; head:=1; tail:=2; t[1]:=a;
   while head<tail do
    begin
     v:=t[head]; tt:=w[v,3];
     while tt<>0 do
      begin
       if dist[v]+w[tt,2]<dist[w[tt,1]]
       then begin
        dist[w[tt,1]]:=dist[v]+w[tt,2];
        if s[w[tt,1]]=0
        then begin s[w[tt,1]]:=1; t[tail]:=w[tt,1]; inc(tail); end;
       end;
       tt:=w[tt,3];
      end;
     inc(head); s[v]:=0;
    end;
   writeln(dist[b]);
  end;
end.