18352_특정 거리의 도시 찾기.py
18352_특정 거리의 도시 찾기.py
문제 요약
N번 도시, M개의 단방향 도로
x 출발 거리가 K인 도시
풀이
deque를 이용한 풀이로 dfs와 유사한 방식으로 풀이
최단경로를 구하는 문제이기 때문에 가장 처음 방문한 경우가 최단이기 때문에 한번 방문한 적이 있으면 더이상 append를 하지 않다.
import sys
from collections import deque
input = sys.stdin.readline
n, m, k, x = map(int ,input().split())
path =[[] for _ in range(n+1)]
dist = [-1] * (n+1)
dist[x] = 0
for _ in range(m):
start,end = map(int, input().split())
path[start].append(end)
deq = deque()
answer = []
deq.append(x)
while deq:
v = deq.popleft()
for i in path[v]:
if dist[i] == -1:
deq.append(i)
dist[i] = dist[v] + 1
for idx, item in enumerate(dist):
if item == k:
answer.append(idx)
answer.sort()
if len(answer):
for i in answer:
print(i)
else:
print('-1')
풀이 2
힙큐를 이용하여 다익스트라 방식으로 통해 풀이.
import sys
import heapq
input = sys.stdin.readline
INF = int(1e9)
answer = []
n, m, k, x = map(int, input().split())
dist = [INF] * (n+1)
graph = [[] for _ in range(n+1)]
for _ in range(m):
a, b = map(int, input().split())
graph[a].append(b)
def function(s):
q = []
heapq.heappush(q, (0, s))
dist[s] = 0
while q:
cost, city = heapq.heappop(q)
if dist[city] < cost:
continue
for i in graph[city]:
if 1 + cost < dist[i]:
dist[i] = 1+cost
heapq.heappush(q, (dist[i], i))
function(x)
for i, v in enumerate(dist):
if v == k:
answer.append(i)
if not answer:
print('-1')
else:
answer.sort()
for i in answer:
print(i)
Author And Source
이 문제에 관하여(18352_특정 거리의 도시 찾기.py), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@kingggyu/18352특정-거리의-도시-찾기.py저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)