[洛谷]P3371 单源最短路径模板 SPFA

#include<cstdio>
#include<cstring>
#include<iostream>


using namespace std;


#define maxn 500005


struct nod{
int to,next,val;
}a[maxn];
int node=0;
int vis[maxn],dis[maxn],head[maxn],queue[maxn];
int n,m,s;


inline void iint() //初始化操作，head[i]=-1 表示这条边已是最后一条边
{
memset(dis,0x3f,sizeof(dis));
for(int i=1;i<=m;i++)
head[i]=-1;
}
inline void add(int u,int v,int w) //建图加边
{
a[++node].to=v;  //这条边指向的点
a[node].val=w;  //这条边的权值
a[node].next=head[u]; //与这条边有共同端点的下一条边
head[u]=node; //道理同上
}
inline void spfa() //spfa算法，广度优先算法
{
int l=0,u,r=1; //l，r分别表示队列的头和尾
queue[1]=s; //第一个点入队
dis[s]=0;/*这个点的最短路径*/ vis[s]=1; //将这个点入队
while(l<r)
{
u=queue[++l]; 
vis[u]=0; //出列操作
for(int i=head[u];i!=-1;i=a[i].next)//遍历以这个点为端点的所有边
{
if(dis[a[i].to]>dis[u]+a[i].val) 
{
dis[a[i].to]=dis[u]+a[i].val;
if(vis[a[i].to]==0) //如果到达的点不在队列中
{
vis[a[i].to]=1;  
queue[++r]=a[i].to;//入队
}
}
}
}
}


int main()
{
scanf("%d%d%d",&n,&m,&s);
iint();
int x,y,z;
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);
}
spfa();
for(int i=1;i<=n;i++)
{
if(dis[i]==0x3f3f3f3f)  //最大值，表示初始点无法到达这个点
cout<<"2147483647 ";
else cout<<dis[i]<<" ";   
}
return 0;
}