BOJ / 2606 / 바이러스 / Python

✍️👀

dfs를 이용했다.

💻 code

import sys
input = sys.stdin.readline

c = int(input())
n = int(input())

graph = [[] for _ in range(c+1)]
visited = [0] * (c+1)

for _ in range(n):
    c1, c2 = map(int, input().split())
    graph[c1].append(c2)
    graph[c2].append(c1)


def dfs(node):
    visited[node] = 1
    for i in graph[node]:
        if not visited[i]:
            dfs(i)


dfs(1)
print(sum(visited)-1)

이때 visited는 True/Fals 가 아닌 1/0 의 값을 갖도록 했다.
따라서 최종 답을 구할 때 sum(visited)를 이용해 방문한 노드들의 개수를 쉽게 구할 수 있다.

Reference