SPFA+Dijkstra 最短路模板

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5,M=1e6+5,INF=2147483647;
int head[N],to[M],nxt[M],dis[N],cost[M];
bool vis[N];
int tot=1;
int n;
void add(int u,int v,int c){
    nxt[++tot]=head[u];
    head[u]=tot;
    to[tot]=v;
    cost[tot]=c;
}
void dijkstra(int s){
    for(int i=1;i<=n;i++) dis[i]=INF;
    dis[s]=0;
    priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>>pq;
    pq.push(make_pair(0,s));
    while(!pq.empty()){
        int u=pq.top().second;
        pq.pop();
        if(vis[u]) continue;
        vis[u]=1;
        for(int i=head[u];i;i=nxt[i]){
            int v=to[i];
            if(!vis[v]&&dis[u]+cost[i]<dis[v]){
                dis[v]=dis[u]+cost[i];
                pq.push(make_pair(dis[v],v));
            }
        }
    }
}
void spfa(int s){
    for(int i=1;i<=n;i++) dis[i]=INF;
    queue<int>que;
    vis[s]=1;
    dis[s]=0;
    que.push(s);
    while(!que.empty()){
        int u=que.front();
        que.pop();
        vis[u]=0;
        for(int i=head[u];i;i=nxt[i]){
            int v=to[i];
            if(dis[v]>dis[u]+cost[i]){
                dis[v]=dis[u]+cost[i];
                if(!vis[v]){
                    vis[v]=1;
                    que.push(v);
                }
            }
        }
    }
}
int main(){
    int m,s;
    scanf("%d %d %d",&n,&m,&s);
    while(m--){
        int u,v,w;
        scanf("%d %d %d",&u,&v,&w);
        add(u,v,w);
    }
    dijkstra(s);
//  spfa(s);
    for(int i=1;i<=n;i++) printf("%d ",dis[i]);
    return 0;
}