알고리즘 스터디 - 백준 2606번 : 바이러스

문제 해석

1번 컴퓨터에 의해 웜 바이러스에 걸리게 되는 컴퓨터의 수를 출력

입력

1. 컴퓨터의 수
2. 컴퓨터 쌍의 개수
3. 연결되어 있는 컴퓨터 번호쌍이 주어짐

알고리즘 코드

computer = int(input())
pair = int(input())

graph = [[] for _ in range(computer+1)]

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

visited = []
queue = [1]

while queue:
    node = queue.pop(0)

    if node not in visited:
        visited.append(node)
        queue.extend(graph[node])

print(len(visited) - 1)