SPFA(链式前向星)

SPFA

与djkstra比可判断负环，处理负权边

//SPFA
#include<bits/stdc++.h>
using namespace std;
const int maxn = 10100;         //最大顶点数
const int maxm = 500500;        //最大边数
const int inf = 0x3f3f3f3f;     //inf无限大
struct edge{                    //链式前向星建图
    int to,next,w;
}e[maxm];
int using_v[maxn],using_times[maxn];//入队的点和点的入队次数
int head[maxn],dis[maxn];
int cnt=0;
int n,m;        //n个点m条边
void addedge(int u,int v,int w){
    e[++cnt].to = v;
    e[cnt].w = w;
    e[cnt].next = head[u];
    head[u] = cnt;
}
int spfa(int start){
    queue<int>q;
    dis[start] = 0;         //起始点离自己的dis为0
    using_v[start] = 1;     //起始点标记
    q.push(start);          //入队
    while(!q.empty()){//fs
        int top=q.front();  //队首
        q.pop();            //取出
        using_v[top] = 0;   //top这个点暂时没在队列中
        using_times[top]++; //使用次数++
        if(using_times[top]>n) return 0;        //存在复环
        for(int i = head[top];i;i=e[i].next){   //图遍历
            if(dis[e[i].to]>dis[top]+e[i].w){   //松弛
                dis[e[i].to] = dis[top]+e[i].w;
                if(!using_v[e[i].to]){
                    using_v[e[i].to] = 1;       //入队
                    q.push(e[i].to);
                }
            }
        }
    }
    return 1;
}
int main()
{
    int u,v,w,s;    //u--->v的权值为w,s为起始点
    cin >> n >> m >> s;
    memset(dis,inf,sizeof(dis));
    for(int i = 0;i < m;i++){
        cin >> u >> v >> w;
        addedge(u,v,w);
    }
    spfa(s);
    for(int i = 1;i <= n;i++){
        if(dis[i] != inf)
            cout << dis[i] << " ";
        else
            cout << "inf" << " ";
    }

    return 0;
}