spfa模版

题目描述

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

输入输出格式

输入格式：

第一行包含三个整数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 <map>
#include <iterator>
#include <algorithm>
#include <vector>
#include <queue>
#include <list>
#include <cmath>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <string>
#include <sstream>
#define INF 1e6
using namespace std;
const int maxn = 10005;
const int maxm = 500005;
int num_edge=0,n,m,src;
int dis[maxn],vis[maxn],head[maxn];
struct Edge{
    int next,to,dis;//next
}edge[maxm];
void addedge(int from,int to,int dis){
    edge[++num_edge].next = head[from];//链式存储下一条出边
    edge[num_edge].to = to;//当前节点编号
    edge[num_edge].dis = dis;
    head[from] = num_edge;//记录下一次出边情况
}
void spfa(){
    queue<int >q;
    memset(dis,INF,sizeof dis);//带权图初始化
    memset(vis,0,sizeof vis);
    q.push(src);
    dis[src] = 0;
    vis[src] = 1;
    while(!q.empty()){
        int u=q.front();
        q.pop();
        vis[u]=0;
        for(int i=head[u];i;i=edge[i].next){
            int v=edge[i].to;
            if(dis[v]>dis[u]+edge[i].dis){
                dis[v]=dis[u]+edge[i].dis;
                if(vis[v]==0){
                    vis[v]=1;
                    q.push(v);
                }
            }
        }
    }
}
int main(){
    int a,b,c;
    cin>>n>>m>>src;
    for(int i=0;i<m;i++){
        cin>>a>>b>>c;
        addedge(a, b, c);
    }
    spfa();
    for(int i=1;i<=n;i++)
        if(src==i)
            cout<<0<<" ";
        else
            cout<<dis[i]<<" ";
    return 0;
}