[백준-6118] 숨바꼭질 / Python

6118번: 숨바꼭질

import sys
import heapq

input = sys.stdin.readline
INF = int(1e9)

n, m = map(int, input().split())
graph = [[] for _ in range(n + 1)]
distance = [-1] * (n + 1)


def dijkstra(start):
    q = []
    heapq.heappush(q, (0, start))
    distance[start] = 0
    while q:
        dist, now = heapq.heappop(q)
        for i in graph[now]:
            if distance[i] == -1:
                distance[i] = dist + 1
                heapq.heappush(q, (distance[i], i))


for _ in range(m):
    a, b = map(int, input().split())
    graph[a].append(b)
    graph[b].append(a)

dijkstra(1)

print(distance.index(max(distance)), max(distance), distance.count(max(distance)))

 

해설

다익스트라로 해결했다. 어렵지 않은 문제였는데 시작 정점을 1로 설정을 해서 100% 틀렸습니다가 떴다..😂 시작 정점을 0으로 설정해줘야 한다.