spfa(最短路问题)

【题目描述】

给定 N 个点和 M 条单向道路，每条道路都连接着两个点，每个点都有自己编号，分别为 1~ N。问你从 S 点出发，到达每个点的最短路径为多少。

【输入描述】

输入第一行包含三个正整数 N, M, S。第 2 到 M+1 行每行包含三个正整数 u, v, w，表示 u→v 之间存在一条距离为 w 的路。1≤N≤5×10^3，1≤M≤5×10^4，1≤ui, vi≤N，0≤wi≤10^9

【输出描述】

输出仅一行，共 N 个数，分别表示从编号 S 到编号为 1~ N 点的最短距离，两两之间用空格隔开。(如果无法到达则输出-1)

#include<bits/stdc++.h>
using namespace std;
const long long INF=0x3f3f3f3f3f3f3f;
const int N=5e3+10;
struct edge{
    int to;
    long long w;
    edge(int tt,long long ww) {to=tt;w=ww;
    }
};
long long dist[N];
bool st[N];
vector<edge>e[N];
void spfa(int s)
{
    memset(dist,0x3f,sizeof dist);
    dist[s]=0;//起点到自己距离为0
    queue<int>q;
    q.push(s);//s入队
    st[s]=true;//起点在队列中
    while(q.size())
    {
        int u=q.front();
        q.pop();
        st[u]=false;//u已经不在队列里
        if(dist[u]==INF) continue;
        for(int i=0;i<e[u].size();i++)//遍历u的邻居
        {
            int v=e[u][i].to;
            long long w=e[u][i].w;
            if(dist[v]>dist[u]+w)
            {
                dist[v]=dist[u]+w;
                if(!st[v])
                {
                    q.push(v);
                    st[v]=true;
                }
            }
        }
    }
}
int main()
{
    int n,m,s;
    cin>>n>>m>>s;
    for(int i=1;i<=m;i++)
    {
        int u,v;long long w;
        cin>>u>>v>>w;
        e[u].push_back(edge(v,w));
    }
    spfa(s);
    for(int i=1;i<=n;i++)
    {
        if(dist[i]==INF) cout<<-1;
        else cout<<dist[i];
        if(i!=n) cout<<" ";
        else cout<<endl;
    }
    return 0;
}