[Baekjoon/python] 웜홀 #1865

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

1865번: 웜홀

문제

때는 2020년, 백준이는 월드나라의 한 국민이다. 월드나라에는 N개의 지점이 있고 N개의 지점 사이에는 M개의 도로와 W개의 웜홀이 있다. (단 도로는 방향이 없으며 웜홀은 방향이 있다.) 웜홀은 시작 위치에서 도착 위치로 가는 하나의 경로인데, 특이하게도 도착을 하게 되면 시작을 하였을 때보다 시간이 뒤로 가게 된다. 웜홀 내에서는 시계가 거꾸로 간다고 생각하여도 좋다.

시간 여행을 매우 좋아하는 백준이는 한 가지 궁금증에 빠졌다. 한 지점에서 출발을 하여서 시간여행을 하기 시작하여 다시 출발을 하였던 위치로 돌아왔을 때, 출발을 하였을 때보다 시간이 되돌아가 있는 경우가 있는지 없는지 궁금해졌다. 여러분은 백준이를 도와 이런 일이 가능한지 불가능한지 구하는 프로그램을 작성하여라.

코드

• 가중치가 음수인 그래프 문제는 벨만-포드 알고리즘 적용
• float('inf')를 사용할 경우 음수를 더해주어도 float('inf')이기때문에 사용하면 안됨

import sys

input = sys.stdin.readline

TC = int(input())

def bellman_ford():
    mdist = [1e9] * (N + 1)
    mdist[1] = 0

    for n in range(1, N + 1):
        for edge in edges:
            s, e, t = edge
            if mdist[s] + t < mdist[e]:
                mdist[e] = mdist[s] + t
                if n == N:
                    return "YES"
        
    return "NO"


for _ in range(TC):
    N, M, W = map(int, input().split())
    edges = []

    # 도로는 방향이 없음
    for _ in range(M):
        S, E, T = map(int, input().split())
        edges.append((S, E, T))
        edges.append((E, S, T))

    # 웜홀은 방향이 있음
    for _ in range(W):
        S, E, T = map(int, input().split())
        edges.append((S, E, -T))

    print(bellman_ford())