Bellman-Ford算法模板（以POj 3259为例）

分手后的思念是犯贱 2022-05-06 06:52 211阅读 0赞

## 题目：[点击打开链接][Link 1]   题意：题目的大意是有F个农场（F组输入数据），每个农场有N个牧场，M条**双向**路径，W个虫洞，虫洞是**单向**的，可以实现时间旅行，返回到以前某个时间。问从某个牧场出发，经过若干路径和虫洞，能不能在自己没有离开出发地时回到出发地，见到自己。   分析：其实就是判断是不是存在负环，用Bellman-Ford算法求就可以了。 当图中存在负权环时，就能够在出发之前回到出发地，见着自己，将虫洞的权值变成其相反数，然后再用Bellman-Ford算法求解就可以了。Bellman-Ford算法入门参考[https://blog.csdn.net/u013445530/article/details/41761687][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;///松弛ｎ－１次以后还可以松弛说明有负环
                    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;
    }

[Link 1]: http://poj.org/problem?id=3259
[https_blog.csdn.net_u013445530_article_details_41761687]: https://blog.csdn.net/u013445530/article/details/41761687