SPFA模板

最新推荐文章于 2024-07-31 15:18:02 发布

原创最新推荐文章于 2024-07-31 15:18:02 发布 · 291 阅读

0 ·

CC 4.0 BY-SA版权

图论专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种用于寻找图中两点间最短路径的SPFA算法实现。通过使用队列和栈来优化Bellman-Ford算法的时间复杂度，SPFA能够高效地解决负权边但不含负环的问题。文章提供了完整的C++代码示例，并通过一个实例演示了如何求解最短路径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#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;
}