본문 바로가기
  • KEEP HUSTLE!
파이썬 코테 준비

[백준 2606] 바이러스 - Python(파이썬)

by 하수군 2021. 3. 9.
import sys


def solution(idx):
    for i in arr[idx]:
        if check[i] == 0:
            check[i] = 1
            solution(i)


N = int(sys.stdin.readline().strip())
arr = {n: [] for n in range(1, N+1)}
D = int(sys.stdin.readline().strip())
check = [0] * (N + 1)

for _ in range(D):
    a, b = map(int, sys.stdin.readline().split())
    arr[a].append(b)
    arr[b].append(a)

check[1] = 1
solution(1)
print(sum(check) - 1)

댓글