Bellman-Ford算法模板(以POj 3259为例)

题目:点击打开链接
题意:题目的大意是有F个农场(F组输入数据),每个农场有N个牧场,M条双向路径,W个虫洞,虫洞是单向的,可以实现时间旅行,返回到以前某个时间。问从某个牧场出发,经过若干路径和虫洞,能不能在自己没有离开出发地时回到出发地,见到自己。
分析:其实就是判断是不是存在负环,用Bellman-Ford算法求就可以了。 当图中存在负权环时,就能够在出发之前回到出发地,见着自己,将虫洞的权值变成其相反数,然后再用Bellman-Ford算法求解就可以了。Bellman-Ford算法入门参考https://blog.csdn.net/u013445530/article/details/41761687

代码:

#pragma comment(linker, "/STACK:102400000,102400000")
///#include<unordered_map>
///#include<unordered_set>
#include<algorithm>
#include<iostream>
#include<fstream>
#include<complex>
#include<cstdlib>
#include<cstring>
#include<cassert>
#include<iomanip>
#include<string>
#include<cstdio>
#include<bitset>
#include<vector>
#include<cctype>
#include<cmath>
#include<ctime>
#include<stack>
#include<queue>
#include<deque>
#include<list>
#include<set>
#include<map>
using namespace std;
#define pt(a) cout<<a<<endl
#define debug test
#define mst(ss,b) memset((ss),(b),sizeof(ss))
#define rep(i,a,n) for (int i=a;i<=n;i++)
#define per(i,a,n) for (int i=n-1;i>=a;i--)
#define pii pair<int,int>
#define fi first
#define se second
#define ll long long
#define ull unsigned long long
#define pb push_back
#define mp make_pair
#define inf 0x3f3f3f3f
#define eps 1e-10
#define PI acos(-1.0)
const ll mod = 1e9+7;
const int N = 1e3+10;

ll qp(ll a,ll b) {ll res=1;a%=mod; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%mod;a=a*a%mod;}return res;}
int to[4][2]={{-1,0},{1,0},{0,-1},{0,1}};

int f,n,m,w,d[N];
struct eg{
    int u,v,w;
};
vector<eg> e;

int bf() {
    ///mst(d,0) 
    ///判负环所有的正权路径都不需要被考虑,所以可以把所有的d都设为0
    ///这样只有负权路径才会被转移
    mst(d,inf);
    d[1]=0;
    int fg;
    for(int i=1;i<=n;i++) {
        fg=1;
        for(int j=0;j<e.size();j++) {
            if(d[e[j].v]>d[e[j].u]+e[j].w) {
                if(i==n) return 1;///松弛n-1次以后还可以松弛说明有负环
                d[e[j].v]=d[e[j].u]+e[j].w;
                fg=0;
            }
        }
        if(fg) return 0;///不能松弛,说明不存在负环
    }
    return 0;
}

int main() {
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    cin>>f;
    while(f--) {
        e.clear();
        cin>>n>>m>>w;
        int a,b,c;
        for(int i=1;i<=m;i++) {
            cin>>a>>b>>c;
            e.pb({a,b,c});
            e.pb({b,a,c});
        }
        for(int i=1;i<=w;i++) {
            cin>>a>>b>>c;
            e.pb({a,b,-c});
        }
        if(bf()) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    return 0;
}

 

    原文作者:Bellman - ford算法
    原文地址: https://blog.csdn.net/tianwei0822/article/details/83037993
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞