본문 바로가기
알고리즘 풀이

[백준 / Python] 24479번: 알고리즘 수업 -깊이 우선 탐색 1 - jini

by developer jini 2022. 7. 5.
728x90

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

 

24479번: 알고리즘 수업 - 깊이 우선 탐색 1

첫째 줄에 정점의 수 N (5 ≤ N ≤ 100,000), 간선의 수 M (1 ≤ M ≤ 200,000), 시작 정점 R (1 ≤ R ≤ N)이 주어진다. 다음 M개 줄에 간선 정보 u v가 주어지며 정점 u와 정점 v의 가중치 1인 양

www.acmicpc.net

<문제풀이>

import sys

sys.setrecursionlimit(10 ** 9)
input1 = sys.stdin.readline

# 차례대로 정점의 수 , 간선의 수 , 시작 정점.
n, m, r = map(int, input1().split())
graph = [[] for _ in range(n + 1)]
for _ in range(m):
    a, b = map(int, input1().split())
    graph[a].append(b)
    graph[b].append(a)

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


# result = [r] # 메모리오류


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


dfs(r)
for i in range(1, n + 1):
    print(visited[i])
    # if i in result: 메모리 초과
    #     print(result.index(i) + 1)
    # else:
    #     print(0)

 

728x90

댓글