> 技术文档 > 图论-最短路 Bellman-Ford算法

图论-最短路 Bellman-Ford算法


文章目录

  • Bellman-Ford算法
    • SPFA优化
    • 路径记录与传输

Bellman-Ford算法

此算法是基于松弛操作的单源最短路算法。
e[u]存u点的出边的邻边和边权,d[u]存u点到源点的距离。

  • 1.初始化,d[s]=0,d[其他点]=INT_MAX。
  • 2.执行多轮循环,对所有边都尝试进行一次松弛操作。
  • 3.当一轮中没有成功的松弛操作时,算法停止。

如图,3为源点
图论-最短路 Bellman-Ford算法
代码展示

struct edug{int v,w;}; vector<edge>e[N];int d[N];bool Bellman-Ford(int s){//把所有点到源点的距离初始为无穷大memset(d,0x3f3f3f,sizeof d);d[s]=0;//源点到自身的距离为0bool f;//标记是否有松弛操作for(int i=1;i<=n;i++)//n轮操作 {f=false;for(int u=1;u<=n;u++)//n个点 {if(d[u]==0x3f3f3f)continue;for(auto x:ed[u])//遍历u的出边 {int v=x.v,w=x.w;if(d[u]+w<d[v]){d[v]=d[u]+w;f=true;}}}if(!f)break;//没有得到更新就退出  } return f;//第n轮=true则有环 }int main(){cin>>n>>m>>s;for(int i=0;i<m;i++){cin>>a>>b>>c;e[a].push_back({b,c});}return 0; } 

优点:
可以正确处理负边权
可以判负环,如果n轮循环后仍存在能松弛的边,则说明存在负环。
时间复杂度:O(nm)

SPFA优化

我们可以想到只有本轮被更新的点,其出边才有可能引起下一轮的松弛操作,因此用队列来维护被更新的点的集合。
vis[u]标记u点是否在队内,cnt[v]记录边数,判负环。

  • 1.初始化,s入队,并标记s在队内,d[s]=0,d[其他点]=INT_MAX
  • 2.从队头弹出u点,标记u不在队内
  • 3.枚举u的所有出边,执行松弛操作,记录s到v的边数,并判负环。如果v不在队内则把v 压入队尾,并标记
  • 重复2,3操作,直到队列为空。

代码展示

struct edug{int v,w;}; vector<edge>e[N];int d[N];queue<int>q;bool spfa(int s){memset(d,0x3f3f3f,sizeof d);d[s]=0,vis[s]=1;q.push(s);while(q.size()){int u=q.front();q.pop();vis[u]=0;for(auto x:e[u]){int v=x.v,w=x.w;if(d[u]+w<d[v]){d[v]=d[u]+w;cnt[v]=cnt[u]+1;if(cnt[v]>=n)return true;if(!vis[v])q.push(v),vis[v]=1;}}}return false;}

路径记录与传输

void dfs_path(int u){if(u==1){cout<<u<<\" \";return ;}dfs_path(pre[u]);cout<<u<<\" \";}