알고리즘

[백준, 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)