알고리즘
[백준, 2606번] 바이러스
hminor
2023. 8. 13. 11:40
반응형
dictionary로 연결된 컴퓨터의 종류를 추가 후
방문했는지를 visit로 체크하며
방문하지 않았다면 결과에 +1하는 형식으로 문제 해결
import sys
from collections import deque
input = sys.stdin.readline
com = int(input())
dic = {i:[] for i in range(1, com+1)}
visit = [0]*(com+1)
visit[1] = 1
result = 0
for _ in range(int(input())):
s,e = map(int,input().split())
dic[s].append(e)
dic[e].append(s)
q = deque([1])
while q:
s = q.popleft()
for i in dic[s]:
if not visit[i]:
visit[i] = 1
result +=1
q.append(i)
print(result)