SPFA模板

我的心愿是世界和平！

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;

#define MAXN 0x3f3f3f3f
int dis[10005],head[10005];
bool sign[10005];
struct edge{
    int u,v,w;
    int next;
}e[10005];
queue<int>q;
int n,m,k,s,t,cnt=1;

void add(int a,int b,int c){
    e[cnt]=((edge){a,b,c,head[a]});
    head[a]=cnt++;
}

void SPFA(){
    memset(dis,MAXN,sizeof(dis));
    memset(sign,0,sizeof(sign));
    dis[s]=0;
    sign[s]=1;
    q.push(s);
    while(!q.empty()){
        int x=q.front();
        q.pop();
        sign[x]=0;
        for(int i=head[x];i!=0;i=e[i].next){
            if(dis[e[i].v]>dis[e[i].u]+e[i].w){
                dis[e[i].v]=dis[e[i].u]+e[i].w;
                if(!sign[e[i].v]){
                    q.push(e[i].v);
                    sign[e[i].v]=1;
                }
            }
        }
    }
}

int main(){
    memset(head,0,sizeof(head));
    scanf("%d%d",&n,&m);//输入n点数，m边数
    scanf("%d%d",&s,&t);//输入s起始点，t终点
    for(int i=1;i<=m;i++){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);//输入a，b两点及其距离c
        add(a,b,c);
        add(b,a,c);
    }
    SPFA();
    for(int i=1;i<=n;i++){
        printf("%d ",dis[i]);
    }
    return 0;
}