2606.바이러스 python

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

2606번: 바이러스

 

bfs 를 사용한 간단하다 

딕셔너리로 어디에서 어디가 이어지는지 만들고 

재귀함수로 방문 가능한곳을 모두 집합에 넣어서 그 집합의 크기가 바이러스 퍼지는 크기다 

d={}
input()
n=int(input())


for i in range(n):
    i=list(map(int,input().split()))
    d.setdefault(i[0],set())
    d.setdefault(i[1],set())
    d[i[0]].add(i[1])
    d[i[1]].add(i[0])
def f(x,o):
    for i in d[x]:
        if i not in o:
            o.add(i)
            f(i,o)
    return o
a=len(f(1,set()))
print( 0 if a<=1 else a-1)