BOJ 2606 바이러스 Python3

 

import sys
input = sys.stdin.readline

computer_cnt = int(input().rstrip())
graph = [[False]*(computer_cnt+1) for _ in range(computer_cnt+1)]
link_cnt = int(input().rstrip())
for _ in range(link_cnt):
    x, y = map(int, input().rstrip().split())
    graph[x][y] = True
    graph[y][x] = True

visited = [1]
result = 0
def dfs(x):
    global result
    warm_list = []
    for i in range(1, computer_cnt+1):
        if graph[x][i] and (i not in visited):
            warm_list.append(i)
            visited.append(i)
            dfs(i)
    result += len(warm_list)

dfs(1)
print(result)

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

2606번: 바이러스