spfa模板

//spfa在正权边题目容易被卡,所以正权边的情况下还是用dijkstra吧
//spfa比dijkstra的优点是可以判负环,处理负权边
//spfa的时间复杂度为(O(|V||E|))
//spfa是求单元最短路
//spfa在最小费用最大流中常用

#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
#include<algorithm>
#include<iostream>
using namespace std;
typedef long long ll;
#define endl '\n'
const int maxn = 10100;//最大顶点数
const int maxm = 500500;//最大边数
const int inf = 0x3f3f3f3f;
struct edge{
    int to,nxt,w;
}e[maxn];

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].nxt=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()){
        int top=q.front();  
        q.pop();            
        using_v[top]=0;     
        if( using_times[top]>n ) return 0;  //存在负环
        for(int i=head[top];~i;i=e[i].nxt){
            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,0x3f,sizeof(dis));
    memset(head,-1,sizeof(head));
    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]<<endl;
        }else{
            cout<<"inf"<<endl;
        }
    }
    return 0;
}