洛谷P3371 单源最短路径（spfa）

【模板】单源最短路径

题目描述

如题，给出一个有向图，请输出从某一点出发到所有点的最短路径长度。

输入输出格式

输入格式：

第一行包含三个整数N、M、S，分别表示点的个数、有向边的个数、出发点的编号。

接下来M行每行包含三个整数Fi、Gi、Wi，分别表示第i条有向边的出发点、目标点和长度。

输出格式：

一行，包含N个用空格分隔的整数，其中第i个整数表示从点S出发到点i的最短路径长度（若S=i则最短路径长度为0，若从点S无法到达点i，则最短路径长度为2147483647）

分析：spfa模版。

代码

const
  maxn=1000000;
var
  ls,s,d,next:array[0..maxn] of int64;
  v:array[0..maxn] of boolean;
  a:array[0..maxn,1..3] of longint;
  i,j,n,m,k:longint;


procedure spfa;
var
  head,tail,i:longint;
begin
  head:=0;
  tail:=1;
  s[1]:=k;
  v[k]:=true;
  while head<tail do
    begin
      inc(head);
      i:=ls[s[head]];
      while i>0 do
        begin
          if d[a[i,1]]+a[i,3]<d[a[i,2]] then
            begin
              d[a[i,2]]:=d[a[i,1]]+a[i,3];
              if not v[a[i,2]] then
                begin
                  v[a[i,2]]:=true;
                  inc(tail);
                  s[tail]:=a[i,2];
                end;
            end;
          i:=next[i];
        end;
      v[s[head]]:=false;
    end;
end;


begin
  readln(n,m,k);
  for i:=1 to n do
    d[i]:=maxlongint;
  d[k]:=0;
  for i:=1 to m do
    begin
      readln(a[i,1],a[i,2],a[i,3]);
      next[i]:=ls[a[i,1]];
      ls[a[i,1]]:=i;
    end;
  spfa;
  for i:=1 to n do
    write(d[i],' ');
end.