Algorithm/boj

[파이썬] 24480 알고리즘 수업 - 깊이 우선 탐색 2

takeU 2022. 6. 27. 11:12
반응형
import sys
sys.setrecursionlimit(10 ** 8)
input = sys.stdin.readline

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

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

def dfs(node):
    global count
    visited[node] = 1
    ans[node] = count
    graph[node].sort(reverse=True)
    for g in graph[node]:
        if visited[g] == 0:
            count += 1
            dfs(g)

dfs(r)

for a in ans[1:]:
    print(a)

dfs