-
Notifications
You must be signed in to change notification settings - Fork 0
/
210305_BOJ_11657.py
43 lines (34 loc) · 1.09 KB
/
210305_BOJ_11657.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import sys
input = sys.stdin.readline
INF = int(1e9)
def bellman_ford(start):
dist[start] = 0
# n번의 라운드를 반복
for i in range(1, n + 1):
# 매 라운드마다 모든 간선을 확인
for j in range(m):
now, next, cost = edges[j][0], edges[j][1], edges[j][2]
# 현재 간선을 거쳐서 다른 노드로 이동하는 거리가 더 짧은 경우
if dist[now] != INF and dist[next] > dist[now] + cost:
dist[next] = dist[now] + cost
# n번째 라운드에서도 값이 갱신된다면 음수 순환 존재
if i == n:
return True
return False
n, m = map(int, input().split())
edges = []
dist = [INF] * (n + 1)
for _ in range(m):
a, b, c = map(int, input().split())
edges.append((a, b, c))
negative_cycle = bellman_ford(1)
if negative_cycle:
print(-1)
else:
for i in range(2, n + 1):
# 도달할 수 없는 경우
if dist[i] == INF:
print(-1)
# 도달 가능한 경우
else:
print(dist[i])