😊 백준 2606 : 바이러스

2606: 바이러스

DFS/BFS 두 번째 문제!

📌 Try 1

def dfs(graph, start_node, visit):
    visit.append(start_node)
    # print(start_node, end=' ')
    
    for node in graph[start_node]:
        if node not in visit:
            dfs(graph, node, visit)

graph = dict()
visit = list()

N = int(input())
nums = int(input())
for _ in range(nums):
    x, y = map(int, input().split())
    
    if x in graph:
        graph[x].append(y)
    else:
        graph[x] = [y]
    if y in graph:
        graph[y].append(x)
    else:
        graph[y] = [x]
        
dfs(graph, 1, visit)

print(len(visit)-1)

한 번에 풀었다~! 간단해 보여서 이전에 작성한 dfs 재귀 코드를 그대로 가져다 사용했다.

🎁 Reference

• dfs 재귀 코드