[BOJ] 24445 | 알고리즘 수업 - 너비 우선 탐색 2

아래 백준 로고를 클릭하면 해당 문제로 이동합니다 😀

풀이과정

---

[BOJ] 24444 | 알고리즘 수업 - 너비 우선 탐색 1과 동일하지만 방문 노드 정렬을 내림차순으로 바꿔주면 된다.

코드를 살~짝 다르게 짜봤다. 굳이 result를 사용하지 않아도 visited에 방문 숫자를 넣어주면 되지않나 ? 라는 생각을 해서 visited만 사용해서 만들어봤다!

코드

---

import sys
from collections import deque
sys.setrecursionlimit(10**8)

N, M, R = map(int, sys.stdin.readline().split())

graph = [[] for _ in range(N+1)]

for _ in range(M):
    u, v = map(int, sys.stdin.readline().split())
    graph[u].append(v)
    graph[v].append(u)
  
for i in range(N+1):
    graph[i].sort(reverse = True)

visited = [0] * (N+1)

def bfs(graph, r):
    global visited
    q = deque([r])

    visited[r] = 1
    t = 2

    while q:
        node = q.popleft()

        for v in graph[node]:
            if visited[v] == 0:
                q.append(v)
                visited[v] = t
                t += 1

bfs(graph, R)

for i in range(1, N+1):
    print(visited[i])

결과

---