SPFA 模板

#include<bits/stdc++.h>
using namespace std;
#define maxn 10005
#define inf 0x3f3f3f3f

struct edge
{
    int from,to,w,next;
 }e[maxn];
 int head[maxn],vis[maxn],dist[maxn];
 int n,m ,t;
 void add(int i,int j,int w)
 {
    e[t].from=i;
    e[t].to=j;
    e[t].w=w;
    e[t].next=head[i];
    head[i]=t++;
  } 

  void spfa(int s)
  {
    queue<int> q;
    for(int i=1;i<=n;i++){
        dist[i]=inf;
      }
      memset(vis,false,sizeof(vis));
      q.push(s);
      dist[s]=0;
      while(!q.empty()){
        int u=q.front();
        q.pop();
        vis[u]=false;
        for(int i=head[u];i!=-1;i=e[i].next){
            int v=e[i].to;
            if(dist[v]>dist[u]+e[i].w){
                dist[v]=dist[u]+e[i].w;
                if(!vis[v]){
                    vis[v]=true;
                    q.push(v);
                  }
              }
          }
      }
  }


int main()
{
    scanf("%d%d",&n,&m);
    t=0;
    memset(head,-1,sizeof(head));
    while(m--){
        int s,t,w;
        scanf("%d%d%d",&s,&t,&w);
        add(s,t,w);
    }
    int start,end;
    scanf("%d%d",&start,&end);
    spfa(start);
    if(dist[end]==inf)cout<<-1<<endl;
    else cout<<dist[end]<<endl;
}

猜你喜欢

转载自blog.csdn.net/qq_40273481/article/details/80821278