[PS] BOJ 11657 타임머신

BOJ-11657 타임머신

https://www.acmicpc.net/problem/11657

본 문제는 시간 C가 음수가 나올 수 있기 때문에, 벨만-포드 알고리즘을 사용해야 한다.

음의 페로가 존재한다면 -1 만 출력하고

1부터 A[i] 까지의 거리를 모두 출력하는데 만일 A[i]가 1의정점과 다른 컴포넌트에 있게되면

무한대의 값을 가지게 된다. 이럴때도 -1을 출력해주면 정답이다.

solution

#include<iostream>
#include<vector>
#include<list>
#include<algorithm>
#define MAXN 100000
using namespace std;
int N, M;
vector<list<pair<int, int> > > G;	//to, weight
void bellman_ford(int pos) {
	vector<int> d(N);
	bool b;
	fill(d.begin(), d.end(), MAXN);
	d[pos] = 0;
	for (int i = 0; i < N; i++){
		b = false;
		for (int j = 0; j < G.size();j++){
			for (auto jt = G[j].begin(); jt != G[j].end(); jt++){
				if (d[j] != MAXN && d[jt->first]>d[j] + jt->second){
					d[jt->first]=d[j] + jt->second;
					b = true;
				}
			}
		}
	}
	if (b){
		cout << -1 << endl;
	}
	else{
		for (auto it = d.begin() + 1; it != d.end(); it++){
			if (*it == MAXN)
				cout << -1 << endl;
			else
				cout << *it << endl;
		}
	}
}
int main() {
	cin >> N >> M;
	G.assign(N, list<pair<int, int> >());
	for (int i = 0; i < M; i++){
		int from;
		pair<int, int> p;
		cin >> from >> p.first >> p.second;
		p.first--;
		G[from - 1].push_back(p);
	}
	bellman_ford(0);
	return 0;
}
Copyright (C) 2016 (KimBom) all rights reserved.