[파이썬] 백준 알고리즘 No.24445 알고리즘 수업 - 너비 우선 탐색 2

나의 코드

import sys
sys.setrecursionlimit(10 ** 6)
input = sys.stdin.readline

n, m, r = map(int, input().split())
graph = [[] for _ in range(n + 1)]
visited = [False] * (n + 1)
result = [0] * (n + 1)
count = 1

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

data = [r]
while data:
    cur = data.pop(0)
    visited[cur] = True
    result[cur] = count
    count += 1

    for i in sorted(graph[cur], reverse=True):
        if not visited[i]:
            visited[i] = True
            data.append(i)

for i in result[1:]:
    print(i)

 

http://noj.am/24444 

24444번: 알고리즘 수업 - 너비 우선 탐색 1

 

너비우선탐색 1 문제에 reverse=True 옵션을 추가하여 내림차순으로 bfs