和小哥哥一起刷洛谷(6) 图论之SPFA算法

关于\(spfa\)

spfa伪代码：

void spfa(s){
    最短路数组全部设为无限大;
    队列 q;
    起点s入队;
    s离s的距离设为零;
    while(队列非空){
        取出队首;弹出队首;
        for(int i=0;i<u的出度;i++){
            v=第i个终点;
            w=到v的权值;
            if(到u点的最短路+w<目前到v点的最短路){//松弛
                更新目前到v点的最短路;
                v入队;
            }
        }
    }
}

P3371 单源最短路径（弱化版）

题目描述：

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

输入输出格式：

输入格式：

第一行包含三个整数N、M、S，分别表示点的个数、有向边的个数、出发点的编号。

接下来M行每行包含三个整数Fi、Gi、Wi，分别表示第i条有向边的出发点、目标点和长度。

输出格式：

一行，包含N个用空格分隔的整数，其中第i个整数表示从点S出发到点i的最短路径长度（若S=i则最短路径长度为0，若从点S无法到达点i，则最短路径长度为2147483647）

样例：

输入：

4 6 1
1 2 2
2 3 2
2 4 1
1 3 5
3 4 3
1 4 4

输出：

0 2 4 3

代码：

#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<stack>
#include<queue>
#include<vector>
using namespace std;
const int NR=100000;
int d[NR]/*最短路径存储列表*/,n,m,s;
vector< pair<int,int> > adj[NR];//一个结构体队列
void spfa(int s){//spfa函数
    memset(d,0x3f,sizeof(d));
    queue<int> q;
    q.push(s);d[s]=0;
    while(!q.empty()){
        int u=q.front();q.pop();
        for(int i=0;i<adj[u].size();i++){
            int v=adj[u][i].first;
            int w=adj[u][i].second;
            if(d[u]+w<d[v]){
                d[v]=d[u]+w;
                q.push(v);
            }
        }
    }
}
int main()
{
    scanf("%d%d%d",&n,&m,&s);
    for(int i=0;i<m;i++){
        int st,en,w;
        scanf("%d%d%d",&st,&en,&w);
        adj[st].push_back(make_pair(en,w));
    }
    spfa(s);
    for(int i=1;i<=n;i++){//输出
        if(d[i]>=1e9)printf("2147483647 ");
        else printf("%d ",d[i]);
    }
    return 0;
}