Algorithm/Problems

99클럽 코테 스터디 4일차 TIL + dfs (깊이우선탐색)

공부좀하시졍 2024. 11. 1. 00:01

https://www.acmicpc.net/problem/24479

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

n, m, r = map(int, input().split())
graph = [[] for _ in range(n+1)]
visited = [0] * (n+1)
seq = 1 # 방문순서

def dfs(r):
    global seq
    visited[r] = seq
    graph[r].sort()
    for i in graph[r]:
        if visited[i] == 0:
            seq += 1
            dfs(i)

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

dfs(r)
for i in range(1, n+1):
    print(visited[i])
  • 정점 u,v 가 양방향 간선 이기 때문에 u와 v에 각각 append 해준다.
  • 방문 순서를 출력해야 하기 때문에 방문순서 seq와 visited[] 를 사용한다.
  • 인접 정점은 오름차순으로 방문해야 하기 때문에 정점 u에 인접한 접점들을 방문하기 전에 정렬해준다.
  • 방문하지 않았다면 방문순서를 더해주며 visited[] 에 초기화 해준다.