【算法】单元最短路径之Bellman-Ford算法和SPFA算法

旧时模样 提交于 2021-02-18 17:54:12

SPFA是经过对列优化的bellman-Ford算法,因此,在学习SPFA算法之前,先学习下bellman-Ford算法。

bellman-Ford算法是一种通过松弛操作计算最短路的算法。

适用条件

1.单源最短路径(从源点s到其它所有顶点v);
2.有向图&无向图(无向图可以看作(u,v),(v,u)同属于边集E的有向图);
3.边权可正可负(如有负权回路输出错误提示);
4.差分约束系统;

bellman-Ford的具体操作是这样的:

  1. 初始化,dis数组表示从起点到达第i个点的最短距离。初始化时:dis[i]=edge[起点][i];如果没有边相接,则设为MAXN;
  2. 循环n-1次,遍历每个边,将dis[边[i].目标节点]=min(dis[边[i].目标节点],dis[边[i].初始节点]+边[i].权值);
  3. 【可选】检验负权回路:判断边集合中的每一条边的两个端点是否收敛。如果存在未收敛的顶点,则算法返回false,表明问题无解;否则算法返回true,并且从源点可达的顶点v的最短距离保存在 dis[v]中。
#include <bits/stdc++.h>
using namespace std;
int n,m;
//Bellman-Ford algorithm
struct Edge
{
    int a,b,v;
}edge[500005];
long long dis[500005];
void bellman_ford(int x)
{
    dis[x]=0;
    for(int i=1;i<=n-1;i++)
    {
        for(int j=1;j<=m;j++)
        {
            dis[edge[j].b]=min(dis[edge[j].b],dis[edge[j].a]+edge[j].v);            
        }
    }
    return ;
    /*
     bool flag = 1; //判断是否含有负权回路  
    for(int i = 1; i <= m; ++i) if(dis[edge[i].b] > dis[edge[i].a] + edge[i].cost)  
        {  
            flag = 0;  
            break;  
        }  
        return flag;  
    */
}
int main()
{
    int s;
    cin>>n>>m>>s;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&edge[i].a,&edge[i].b,&edge[i].v);
    }
    for(int i=1;i<=n;i++)dis[i]=2147483647;
    bellman_ford(s);
    for(int i=1;i<=n;i++)
        cout<<dis[i]<<" ";
    return 0;
}

这个算法在luogu上70分,TLE了两个测试点,因此需要学习一个更牛逼的算法,SPFA,听说这是一个交大的教授发明的。附上原论文连接:https://wenku.baidu.com/view/df249954d4d8d15abe234eff.html

SPFA算法的全称是:Shortest Path Faster Algorithm,是西南交通大学段凡丁于 1994 年发表的论文中的名字。不过,段凡丁的证明是错误的,且在 Bellman-Ford 算法提出后不久(1957 年 )已有队列优化内容,所以国际上不承认 SPFA 算法是段凡丁提出的。

为了避免最坏情况的出现,在正权图上应使用效率更高的Dijkstra算法若给定的图存在负权边,类似Dijkstra算法等算法便没有了用武之地,SPFA算法便派上用场了。

简洁起见,我们约定加权有向图G不存在负权回路,即最短路径一定存在。用数组d记录每个结点的最短路径估计值,而且用邻接表来存储图G。我们采取的方法是动态逼近法:设立一个先进先出的队列用来保存待优化的结点,优化时每次取出队首结点u,并且用u点当前的最短路径估计值对离开u点所指向的结点v进行松弛操作,如果v点的最短路径估计值有所调整,且v点不在当前的队列中,就将v点放入队尾。这样不断从队列中取出结点来进行松弛操作,直至队列空为止。
 
定理:只要最短路径存在,上述SPFA算法必定能求出最小值。证明:每次将点放入队尾,都是经过松弛操作达到的。换言之,每次的优化将会有某个点v的最短路径估计值d[v]变小。所以算法的执行会使d越来越小。由于我们假定图中不存在负权回路,所以每个结点都有最短路径值。因此,算法不会无限执行下去,随着d值的逐渐变小,直到到达最短路径值时,算法结束,这时的最短路径估计值就是对应结点的最短路径值。
#include <bits/stdc++.h>
using namespace std;
int n,m,start;
long long dis[1000005],used[1000005],head[1000005];
int num_edge=0;
struct Graph
{
    int to,w,next;
}graph[10000005];
bool spfa(int x)
{
    queue <int> q;
    dis[x]=0;
    used[x]=1;
    q.push(x);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        used[u]=0;
        for(int i=head[u];i;i=graph[i].next)
        {
            int v=graph[i].to;
            long long Songchi=dis[u]+graph[i].w;
            if(dis[v]>Songchi)
            {
                dis[v]=Songchi;
                if(used[v]==0)
                {
                    used[v]=1;
                    q.push(v);
                }
            }
        }
    }
    return 1;
}
void addedge(int from,int to,int dis)//邻接表存边
{
    graph[++num_edge].next=head[from];
    graph[num_edge].to=to;
    graph[num_edge].w=dis;
    head[from]=num_edge;
 } 
int main()        
{
    memset(head,0,sizeof(head));
    memset(used,0,sizeof(used));
    cin>>n>>m>>start;
    for(int i=1;i<=n;i++)dis[i]=2147483647;
    for(int i=1;i<=m;i++)
    {
        int a,b,w;
        scanf("%d%d%d",&a,&b,&w);
        addedge(a,b,w);
    }
    if(spfa(start))
    {
        for(int i=1;i<=n;i++)
        cout<<dis[i]<<" ";
    }
}

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!