洛谷 P1144 最短路计数

题目描述

给出一个N个顶点M条边的无向无权图，顶点编号为1～N。问从顶点1开始，到其他每个点的最短路有几条。

输入输出格式

输入格式：

输入第一行包含2个正整数N，M，为图的顶点数与边数。

接下来M行，每行两个正整数x, y，表示有一条顶点x连向顶点y的边，请注意可能有自环与重边。

输出格式：

输出包括N行，每行一个非负整数，第i行输出从顶点1到顶点i有多少条不同的最短路，由于答案有可能会很大，你只需要输出mod 100003后的结果即可。如果无法到达顶点i则输出0。

输入输出样例

输入样例#1：

5 7

1 2

1 3

2 4

3 4

2 3

4 5

4 5

输出样例#1：

1

1

1

2

4

说明

1到5的最短路有4条，分别为2条1-2-4-5和2条1-3-4-5（由于4-5的边有2条）。

对于20%的数据，N ≤ 100；

对于60%的数据，N ≤ 1000；

对于100%的数据，N<=1000000，M<=2000000。

做法：spfa模板

代码如下：

const
maxn=4000000;
type
rec=record
x,y,next:longint;
end;
var
g:array[1..maxn]of rec;
list,ls:array[1..maxn div 2]of longint;
dis,ans:array[1..maxn div 2]of int64;
v:array[1..maxn div 2]of boolean;
n,m:longint;

procedure cin(i,j,k:longint);
begin
with g[k] do
begin
x:=i; y:=j;
next:=ls[x];
ls[x]:=k;
end;
end;

procedure init;
var
i,k,q,p:longint;
begin
k:=0;
readln(n,m);
for i:=1 to m do
begin
inc(k);
readln(p,q);
cin(p,q,k);
inc(k);
cin(q,p,k);
end;
end;

procedure spfa;
var
head,tail,t:longint;
begin
head:=0;
tail:=1;
ans[1]:=1;
dis[1]:=1;
list[1]:=1;
while head<tail do
begin
inc(head);
t:=ls[list[head]];
while t>0 do
with g[t] do
begin
if ans[y]=0 then
begin
v[y]:=true;
ans[y]:=ans[x];
dis[y]:=(dis[x]+1) mod 100003;
inc(tail);
list[tail]:=y;
end
else if dis[x]+1=dis[y]  then  ans[y]:=(ans[x]+ans[y]) mod 100003;
t:=next;
end;
v[list[head]]:=false;
end;
end;

procedure print;
var
i:longint;
begin
for i:=1 to n do
writeln(ans[i]);
end;

begin
init;
spfa;
print;
end.