[백준 2307] 도로검문

Junyoung Park·2022년 3월 13일
0

코딩테스트

목록 보기
254/631
post-thumbnail

1. 문제 설명

도로검문

2. 문제 분석

비활성화할 수 있는 간선은 최대 1개로, 최단 경로를 구성하는 간선을 비활성화해야 거리를 바꿀 수 있다. 다익스트라와 BFS를 통해 최단 경로를 구성하는 간선을 카운트할 수 있고, 하나씩 비활성화 -> 활성화하면서 한 번씩 다익스트라를 돌려 각 간선을 비활성화했을 때 구해지는 기존 최단 경로와의 차의 최댓값을 리턴한다.

3. 나의 풀이

import sys
import heapq
from collections import deque

INF = sys.maxsize
n, m = map(int, sys.stdin.readline().rstrip().split())
nodes = [[] for _ in range(n+1)]

for _ in range(m):
    a, b, t = map(int, sys.stdin.readline().rstrip().split())
    nodes[a].append([b, t])
    nodes[b].append([a, t])

def Dijkstra():
    distances = [INF for _ in range(n+1)]
    distances[1] = 0

    pq = []
    heapq.heappush(pq, [0, 1])

    while pq:
        cur_cost, cur_node = heapq.heappop(pq)

        if distances[cur_node] < cur_cost: continue

        for next_node, next_cost in nodes[cur_node]:
            if distances[next_node] > next_cost + cur_cost:
                distances[next_node] = next_cost + cur_cost
                heapq.heappush(pq, [next_cost+cur_cost, next_node])

    return distances

distances = Dijkstra()
edges = set()

def BFS():
    queue = deque()
    queue.append(n)

    while queue:
        cur_node = queue.popleft()

        if cur_node == 1: continue

        for post_node, post_cost in nodes[cur_node]:
            if distances[post_node] + post_cost == distances[cur_node] and tuple((post_node, cur_node, post_cost)) not in edges:
                edges.add(tuple((post_node, cur_node, post_cost)))
                queue.append(post_node)

BFS()

original = distances[n]
local_diff = 0
for edge in edges:
    a, b, t = edge
    nodes[a].remove([b, t])
    nodes[b].remove([a, t])
    distances = Dijkstra()
    if distances[n] == INF: break
    local_diff = max(local_diff, distances[n]-original)
    nodes[a].append([b, t])
    nodes[b].append([a, t])

if distances[n] == INF: print(-1)
else: print(local_diff)
profile
JUST DO IT

0개의 댓글