POJ 3259 Wormholes 最短路+负环
原题链接:http://poj.org/problem?id=3259
题意
有个很厉害的农民,它可以穿越虫洞去他的农场,当然他也可以通过道路,虫洞都是单向的,道路都是双向的,道路会花时间,虫洞会倒退时间,问你这个农民可不可以回到起点,并且在他出发的时间之前。
题解
就虫洞用负边链接,然后判断是否有负环即可。
代码
#include<iostream> #include<cstring> #include<vector> #include<string> #include<algorithm> #include<queue> #define INF 2500003 #define MAX_N 555 using namespace std; struct edge { public: int to, cost; edge(int t, int c) : to(t), cost(c) { } edge() { } }; vector<edge> G[MAX_N]; int d[MAX_N]; queue<int> que; bool inQue[MAX_N]; bool vis[MAX_N]; int n,m,w; int cnt[MAX_N]; void init(){ for(int i=0;i<=n;i++)G[i].clear(); fill(d,d+n+1,INF); memset(vis,0,sizeof(vis)); while(que.size())que.pop(); memset(inQue,0,sizeof(inQue)); memset(cnt,0,sizeof(cnt)); } bool spfa(int s) { que.push(s); inQue[s] = 1; vis[s] = 1; d[s] = 0; cnt[s] = 1; while (que.size()) { int u = que.front(); que.pop(); inQue[u] = 0; for (int i = 0; i < G[u].size(); i++) { int v = G[u][i].to; int c = G[u][i].cost; if (d[u] + c < d[v]) { d[v] = d[u] + c; if (!inQue[v]) { inQue[v] = 1; que.push(v); cnt[v]++; if (cnt[v] > n)return true; } } } } return false; } int T; int main() { cin.sync_with_stdio(false); cin >> T; while (T--) { cin >> n >> m >> w; init(); for (int i = 0; i < m; i++) { int u, v, c; cin >> u >> v >> c; G[u].push_back(edge(v, c)); G[v].push_back(edge(u, c)); } for (int i = 0; i < w; i++) { int u, v, c; cin >> u >> v >> c; G[u].push_back(edge(v, -c)); } bool flag = false; for (int i = 0; i < n; i++) if (!vis[i]) { flag = spfa(i); if (flag) break; } if (flag)cout << "YES" << endl; else cout << "NO" << endl; } return 0; }