HDU 2544 最短路 SPFA模板题

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2544

题意&题解

最短路模板题,以此题记录SPFA模板

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;

#define INF 0x3f3f3f3f
const int maxn=105;
const int maxm=10005;
int visit[maxn],dist[maxn];
int n,m,id;

//链式前向星建图
struct Node
{
int v,w,next;
}edges[maxm];
int head[maxn];

void addedge(int u,int v,int w) //u到v的权值为w的边
{
edges[id].v=v;
edges[id].w=w;
edges[id].next=head[u]; //把之前的第一条边作为当前边的最后一条边
head[u]=id++; //id为从0开始的标号
}

//建图
void init()
{
memset(head,-1,sizeof(head));
int u,v,w;
id=0;
for(int i=0;i<m;i++)
{
scanf("%d %d %d",&u,&v,&w);
addedge(u,v,w);
addedge(v,u,w);
}
}

void spfa(int st)
{
fill(visit,visit+maxn,0); //标记所有节点未被访问
fill(dist,dist+maxn,INF); //求最短路,初始化最大值
queue<int> Q;
visit[st]=1;
dist[st]=0;
Q.push(st);
while (!Q.empty())
{
int now=Q.front();
Q.pop();
visit[now]=0; //注意此处将该点标记为未访问
for(int i=head[now];i!=-1;i=edges[i].next)
{
int v=edges[i].v;
int w=edges[i].w;
if(dist[v]>dist[now]+w) //最短路松弛
{
dist[v]=dist[now]+w;
if(!visit[v])
{
Q.push(v);
visit[v]=1;
}
}
}
}
}

void solve()
{
while(scanf("%d %d",&n,&m)!=EOF)
{
if(n==0 &&m==0) break;
init();
spfa(1);
int ans=dist[n];
printf("%d\n",ans);
}
}

int main()
{
freopen("input.txt","r",stdin);
solve();
return 0;
}