Bellman_ford算法

申明

再次申明这个专栏只是分享博主学习图论时的顺序，给新手提供一个入门的好方法（博主现在也是新手,所以博主写的代码适合C转C++的新手学习），具体算法逻辑需要自己学习，代码区域有部分解释,因此有错误可以私信或者直接在评论区骂。

前言

Bellman_ford适用于有负权边的最短路问题，时间复杂度为Onm。

题目链接

P3385 【模板】负环 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

注意该题是求的是从一可以达到的地方是否有负环！！！

代码

#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
#include<string>
using namespace std;
#define ll long long
typedef pair<int,int>PII;
const int N=2e5+10,M=3e5+10;
int t,n,m,idx;
int mem[N];
int dist[N];
bool check;
struct node
{
  int a,b,c;  
}edge[M];

void Bellman_ford()
{
    memset(dist,0x3f,sizeof dist);
    dist[1]=0;
    for(int i=1;i<=n-1;i++)
    {
       memcpy(mem,dist,sizeof dist);
       for(int j=1;j<=idx;j++)
       {
           int a=edge[j].a,b=edge[j].b,c=edge[j].c;
           if(mem[a]!=0x3f3f3f3f&&dist[b]>mem[a]+c)  //关键
           {
               dist[b]=mem[a]+c;
           }
       }
    }
       for(int j=1;j<=idx;j++)
       {
           int a=edge[j].a,b=edge[j].b,c=edge[j].c;
           if(dist[a]==0x3f3f3f3f || dist[b]==0x3f3f3f3f) continue; //关键
           if(dist[b]>dist[a]+c) check=true;
       }
}

void solve()
{
    check=false;
    idx=0;
    memset(edge,0,sizeof edge);
    scanf("%d %d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        scanf("%d %d %d",&u,&v,&w);
    if(w>=0) 
    {
        edge[++idx]={u,v,w};
        edge[++idx]={v,u,w};
    }
     else
     {
         edge[++idx]={u,v,w};
      }
    }
     Bellman_ford();
    if(check) printf("YES\n");
    else printf("NO\n");
}

int main()
{
    scanf("%d",&t);
    while(t--) solve();
    return 0;
}

拓展

那么是全局是否有负环代码是什么样的呢，就是把我上面做注释的两个地方修改了一下，如下

代码

#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
#include<string>
using namespace std;
#define ll long long
typedef pair<int,int>PII;
const int N=2e5+10,M=3e5+10;
int t,n,m,idx;
int mem[N];
int dist[N];
bool check;
struct node
{
  int a,b,c;  
}edge[M];

void Bellman_ford()
{
    memset(dist,0x3f,sizeof dist);
    dist[1]=0;
    for(int i=1;i<=n-1;i++)
    {
       memcpy(mem,dist,sizeof dist);
       for(int j=1;j<=idx;j++)
       {
           int a=edge[j].a,b=edge[j].b,c=edge[j].c;
           if(dist[b]>mem[a]+c)  
           {
               dist[b]=mem[a]+c;
           }
       }
    }
       for(int j=1;j<=idx;j++)
       {
           int a=edge[j].a,b=edge[j].b,c=edge[j].c;
           if(dist[b]>dist[a]+c) check=true;
       }
}

void solve()
{
    check=false;
    idx=0;
    memset(edge,0,sizeof edge);
    scanf("%d %d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        scanf("%d %d %d",&u,&v,&w);
    if(w>=0) 
    {
        edge[++idx]={u,v,w};
        edge[++idx]={v,u,w};
    }
     else
     {
         edge[++idx]={u,v,w};
      }
    }
     Bellman_ford();
    if(check) printf("YES\n");
    else printf("NO\n");
}

int main()
{
    scanf("%d",&t);
    while(t--) solve();
    return 0;
}