kuangbin专题专题四 Wormholes POJ

题目链接：https://vjudge.net/problem/POJ-3259

思路：求有无负环，起点随意选就可以，因为目的只是找出有没有负环，有了负环就可以让时间一直回退，那么一定能回到当初，这里我选择从1号点开始。

---

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <string>
#include <vector>
using namespace std;
 
typedef long long LL;
#define inf (1LL << 30) - 1
#define rep(i,j,k) for(int i = (j); i <= (k); i++)
#define rep__(i,j,k) for(int i = (j); i < (k); i++)
#define per(i,j,k) for(int i = (j); i >= (k); i--)
#define per__(i,j,k) for(int i = (j); i > (k); i--)

const int N = 510;
int dis[N];
int n,m,t;
int u,v,w;

struct node{
    int u,v,w;
};

vector<node> E;

void init(){
    memset(dis,inf,sizeof(dis));
    E.clear();
}

void input(){

    rep(i,1,m){

        cin >> u >> v >> w;
        E.push_back(node{u,v,w});
        E.push_back(node{v,u,w});
    }

    rep(i,1,t){

        cin >> u >> v >> w;
        E.push_back(node{u,v,-w});
    }
}

bool bellman_ford(){

    dis[1] = 0;
    bool flag = true;
    rep(i,2,n){

        flag = false;//判断一层循环中有没有出现可以更新的点
        for(int o = 0; o < E.size(); o++){
            if(dis[E[o].v] > dis[E[o].u] + E[o].w)
                dis[E[o].v] = dis[E[o].u] + E[o].w;

            flag = true;//还有可以更新的点
        }

        if(!flag) break;//不存在可以更新的点了，直接退出
    }
    
    //判断有无负环出现，即时间能不能减少
    for(int o = 0; o < E.size(); o++){
        if(dis[E[o].v] > dis[E[o].u] + E[o].w) return true;
    }

    return false;
}

int main(){
 
    ios::sync_with_stdio(false);
    cin.tie(0);

    int T;
    cin >> T;
    while (T--){
        cin >> n >> m >> t;
        
        init();
        input();
        if(bellman_ford()) cout << "YES" << endl;
        else cout << "NO" << endl;
    }

    getchar();getchar();
    return 0;
}