anonymous No title
No License C++
2019年06月08日
Copy Clone

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;

}

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.