【算法模板】邻接表、队列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
#include<bits/stdc++.h>
using namespace std;
#define inf 2147483647
long long n,d[500001],h[500001],num;
struct nope{
int e;
int v;
int next;
}ed[500001];
inline void add(int b,int v,int e)
{
num++;
ed[num].v=v;
ed[num].e=e;
ed[num].next=h[b];
h[b]=num;
return;
}
void spfa(int s)
{
bool l[500001];
queue<int>q;
memset(l,0,sizeof(l));
for(int i=1;i<=n;i++)d[i]=inf;
d[s]=0;l[s]=1;q.push(s);
while(!q.empty())
{
int b=q.front();q.pop();l[b]=0;
for(int i=h[b];i>=0;i=ed[i].next)
{
int e=ed[i].e,v=ed[i].v;
if(d[e]>d[b]+v)
{
d[e]=d[b]+v;
if(!l[e])
{
l[e]=1;
q.push(e);
}
}
}
}
for(int i=1;i<=n;i++)
cout<<d[i]<<" ";
}
int main()
{
int f,g,w,s,m;
memset(h,-1,sizeof(h));
cin>>n>>m>>s;
for(int i=1;i<=m;i++)
{
cin>>f>>g>>w;
add(f,w,g);
}
spfa(s);
}