图论--最短路--bellman-ford 算法

2024年6月9日 108次阅读

图论最短路–bellman ford 算法

** 单源最短路，能判负环。**

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define mem(a,b) memset(a,b,sizeof(a))
#define For(a,b) for(int a=0;a<b;a++)
using namespace std;
const int maxn =  1e4;
const int INF = 0x3f3f3f3f;
const int inf = 0x3f;

int dis[maxn];
struct edge{
    int s,e;    ///起点，终点
    int w;      ///权值
}e[maxn];
int n,m;                ///n为点,m为边的总数
bool bellman(int a,int n)     ///求a->其他点的最短路,n为结点总数.可判负环
{
    memset(dis,inf,sizeof(dis));
    dis[a]=0;
    for(int i=0;i<n-1;i++){              /// VE n*m
        for(int j=0;j<m;j++){
            dis[e[j].e]=min(dis[e[j].e],dis[e[j].s]+e[j].w);    ///松弛操作
        }
    }
    for(int i=0;i<m;i++){                           ///松弛完后还能再松弛即代表有负环
        if(dis[e[i].e]>dis[e[i].s]+e[i].w) return true;
    }
    return false;
}
int main()
{
    cin>>n>>m;
    For(i,m)   /// m 条边
        cin>>e[i].s>>e[i].e>>e[i].w;

    if(bellman(1,n))
        cout << "有负环" << endl;
    else
        for(int i=1;i<=n;i++){
        if(dis[i]!=INF)
            cout<< dis[i] << endl;
        else
           cout <<"INF" << endl;
    }
}
点赞