
No License
C++
2019年06月08日
template< typename T >
struct edge {
int from, to;
T cost;
edge(int to, T cost) : from(-1), to(to), cost(cost) {}
edge(int from, int to, T cost) : from(from), to(to), cost(cost) {}
edge &operator=(const int &x) {
to = x;
return *this;
}
operator int() const { return to; }
};
template< typename T >
using Edges = vector< edge< T > >;
template< typename T >
vector< T > bellman_ford(Edges< T > &edges, int V, int s) {
const auto INF = numeric_limits< T >::max();
vector< T > dist(V, INF);
dist[s] = 0;
for (int i = 0; i < V - 1; i++) {
for (auto &e : edges) {
if (dist[e.from] == INF) continue;
dist[e.to] = min(dist[e.to], dist[e.from] + e.cost);
}
}
return dist;
}
//到達できるできない関係なしに、負の閉路があればtrueを返す
template< typename T >
bool find_negative_loopA(Edges< T > &edges, vector<T> &d, int V) {
//距離の重みを0にしておくことで、始点から順番にたどるという行動をスキップして
//単に構造として負の閉路をもっているかを調べられる
vector< T > dist(V, 0);
for (int i = 0; i < V; i++) {
for (auto &e : edges) {
if (d[e.to] > d[e.from] + e.cost) {
d[e.to] = d[e.from] + e.cost;
//負の閉路がなければV-1で終わる
//負の閉路があるとV回目でも更新が起こる
if (i == V - 1)return true;
}
}
}
return false;
}
//到達できる閉路があるかどうか
//稼いだあとにゴールできるかは問わない
//始点→終点→負の閉路→×終点のパターンはこの関数では検出できない
template< typename T >
bool find_negative_loopB(Edges< T > &edges, vector<T> &d, int V, int s) {
for (int i = 0; i < V; i++) {
for (auto &e : edges) {
if (d[e.from] != INF && d[e.to] > d[e.from] + e.cost) {
d[e.to] = d[e.from] + e.cost;
//負の閉路がなければV-1で終わる
//負の閉路があるとV回目でも更新が起こる
if (i == V - 1)return true;
}
}
}
return false;
}
//始点から負の閉路に到達できて、かつ
//終点が負の閉路に含まれているパターン
//稼いだあとにゴールできる
template< typename T >
bool find_negative_loopC(Edges< T > &edges, vector<T> &d, int V, int s, int t) {
//まず、V-1回の更新のあとに更新が起きてると負の閉路がある。
//この閉路に終点が含まれていると無限に稼げてしまう
//一つの大きな輪になってるとさらにV回かかる(これが最悪パターン)
//V-1以降に更新が起こった時に、辺の行き先toが終点だったら
//負の閉路に終点が含まれることになる。
for (int i = 0; i < 2 * V; i++) {
for (auto &e : edges) {
if (d[e.from] != INF && d[e.to] > d[e.from] + e.cost) {
d[e.to] = d[e.from] + e.cost;
//負の閉路がなければV-1で終わる
//負の閉路があるとV回目でも更新が起こる
if (i >= V - 1 && e.to == t)return true;
}
}
}
return false;
}
//if (find_negative_loopB(es, dist, V, R)) {
// cout << "NEGATIVE CYCLE" << endl;
//}
//else {
// for (int i = 0; i < V; i++) {
//
// if (dist[i] != INF) {
// cout << dist[i] << endl;
// }
// else {
// cout << "INF" << endl;
// }
//
// }
//}
//負の閉路がないとわかっているパターン
//vector< T > bellman_ford(Edges< T > &edges, int V, int s)
//A 到達できるできない関係なしに、負の閉路があればtrueを返す
//負の閉路のチェックのみ(距離は無効)
//bool find_negative_loopA(Edges< T > &edges, vector<T> &dist, int V)
//B
//到達できる閉路があるかどうか
//稼いだあとにゴールできるかは問わない
//始点→終点→負の閉路→×終点のパターンはこの関数では検出できない
//template< typename T >
//bool find_negative_loopB(Edges< T > &edges, vector<T> &dist, int V, int s)
//C
//始点から負の閉路に到達できて、かつ
//終点が負の閉路に含まれているパターン
//稼いだあとにゴールできる
//bool find_negative_loopC(Edges< T > &edges, vector<T> &dist, int V, int s, int t)
int main() {
int V, E, R;
cin >> V >> E >> R;
Edges<int> es;
vector<int> x(E);
vector<int> y(E);
vector<int> z(E);
for (int i = 0; i < E; i++) {
cin >> x[i] >> y[i] >> z[i];
x[i]--; y[i]--;
es.emplace_back(x[i], y[i], z[i]);
es.emplace_back(y[i], x[i], z[i]);
}
vector<int> dist;
//たどり着けない最大値を指定する(10^9で間に合わなかったら書き換える)
dist.resize(V, INF);
dist[R] = 0;
//上のコメントに関数があるので適宜選んで使う
if (find_negative_loopC(es, dist, V, s, t)) {
cout << "NEGATIVE CYCLE" << endl;
}
else {
for (int i = 0; i < V; i++) {
if (dist[i] != INF) {
cout << dist[i] << endl;
}
else {
cout << "INF" << endl;
}
}
}
return 0;
}
No one still commented. Please first comment.