2606 바이러스(Python)

1. 입력을 받고
2. 방문리스트와 그래프리스트를 따로 만들어줬다.
3. BFS 함수로 큐를 이용해 방문처리된 개수를 출력하도록 했다.
4. 그리고 시작 노드(1)을 빼주어 출력했다.

from collections import deque

def bfs(visit, graph, node):
    q = deque([node])
    visit[node] = True

    while q:
        v=q.popleft()
        for i in graph[v]:
            if not visit[i]:
                q.append(i)
                visit[i] = True

n = int(input())
m = int(input())

visit = [False] * (n+1)
graph = [[] for _ in range(n+1)]

for i in range(m):
    a, b = map(int, input().split())
    graph[a].append(b)
    graph[b].append(a)

bfs(visit, graph, 1)
print(visit.count(True)-1)