SPFA模板

最新推荐文章于 2022-01-07 11:50:01 发布

unknown_coder

最新推荐文章于 2022-01-07 11:50:01 发布

阅读量261

点赞数

分类专栏：图论

本文链接：https://blog.csdn.net/ivy_uu/article/details/54884936

版权

图论专栏收录该内容

3 篇文章 0 订阅

订阅专栏

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int cnt,st,ed,s,t,w,h[10005],dis[10005],stack[10005];
int q[40005];
struct Node{
    int to,w,next;
}edge[10005];//10005,1005
void add(int s,int t,int v)
{
    cnt++;
    edge[cnt].to=t;
    edge[cnt].w=v;
    edge[cnt].next =h[s];
    h[s]=cnt;
}
int spfa()
{
    int head=1,tail=1;
    q[1]=st;
    while(head<=tail)
    {
        int u=q[head];
        for(int i=h[u];i;i=edge[i].next)
        {
            int v=edge[i].to;
            if(edge[i].w+dis[u]<dis[v])
            {
                dis[v]=edge[i].w+dis[u];
                if(!stack[v])
                {
                    stack[v]=1;
                    tail++;
                    q[tail]=v;  
                }
            }
        }
        head++; 
        stack[u]=0;//退出栈 
    }
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&s,&t,&w);
        add(s,t,w);
    }
    scanf("%d%d",&st,&ed);//起点，终点 
    memset(dis,1,sizeof dis);
    dis[st]=0;
    spfa();
    cout<<dis[ed]<<endl;
}