[백준] 24444번 알고리즘 수업 - 너비 우선 탐색 1

거북이·2023년 2월 25일
0

백준[실버2]

목록 보기
45/81
post-thumbnail

💡문제접근

  • BFS의 기본 코드 + 별도로 오름차순 정렬 수행
  • 방문 순서를 출력한다는 점을 주의할 것

💡코드(메모리 : 59416KB, 시간 : 636ms)

from collections import deque
import sys
input = sys.stdin.readline

N, M, R = map(int, input().strip().split())
graph = [[] for _ in range(N+1)]
for _ in range(M):
    u, v = map(int, input().strip().split())
    graph[u].append(v)
    graph[v].append(u)

# 오름차순 정렬 수행
for i in range(len(graph)):
    graph[i].sort()

visited = [0] * (N+1)
cnt = 1

def BFS(x):
    global cnt
    queue = deque()
    queue.append(x)
    visited[x] = 1
    while queue:
        v = queue.popleft()
        for i in graph[v]:
            if not visited[i]:
                cnt += 1
                visited[i] = cnt
                queue.append(i)

BFS(R)
for i in visited[1:]:
    print(i)

💡소요시간 : 7m

0개의 댓글