洛谷 P3371 【模板】单源最短路径（弱化版） 题解（SPFA）

题目描述

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

输入格式

第一行包含三个整数 n,m,s，分别表示点的个数、有向边的个数、出发点的编号。

接下来 m 行每行包含三个整数 u,v,w，表示一条 u→v 的，长度为 w 的边。

输出格式

输出一行 n 个整数，第 i 个表示 s 到第 i 个点的最短路径，若不能到达则输出 2^{31}-1。

AC代码 (SPFA)

#include<bits/stdc++.h>
using namespace std;

int n,m,s,x,y,v,cnt;
int d[1000001],vis[1000001],head[1000001],to[1000001],next[1000001],edge[1000001];
queue<int> q;

void add(int x,int y,int v)
{
    cnt++;
    to[cnt]=y;
    edge[cnt]=v;
    next[cnt]=head[x];
    head[x]=cnt;
}

void SPFA()
{
    for (int i=1;i<=n;i++)
        d[i]=2147483647;
    q.push(s);
    d[s]=0;
    vis[s]=1;
    while (q.size())
    {
        int x=q.front();
        q.pop();
        vis[x]=0;
        for (int i=head[x];i;i=next[i])
        {
            int y=to[i],v=edge[i];
            if (d[y]>d[x]+v)
            {
                d[y]=d[x]+v;
                if (vis[y]==0)
                {
                    vis[y]=1;
                    q.push(y);
                }
            }
        }
    }
}

int main()
{
    scanf("%d%d%d",&n,&m,&s);
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&v);
        add(x,y,v);
    }
    
    SPFA();
    
    for (int i=1;i<=n;i++)
        printf("%d ",d[i]);
    return 0;
}