문제
신종 바이러스인 웜 바이러스는 네트워크를 통해 전파된다. 한 컴퓨터가 웜 바이러스에 걸리면 그 컴퓨터와 네트워크 상에서 연결되어 있는 모든 컴퓨터는 웜 바이러스에 걸리게 된다.
예를 들어 7대의 컴퓨터가 <그림 1>과 같이 네트워크 상에서 연결되어 있다고 하자. 1번 컴퓨터가 웜 바이러스에 걸리면 웜 바이러스는 2번과 5번 컴퓨터를 거쳐 3번과 6번 컴퓨터까지 전파되어 2, 3, 5, 6 네 대의 컴퓨터는 웜 바이러스에 걸리게 된다. 하지만 4번과 7번 컴퓨터는 1번 컴퓨터와 네트워크상에서 연결되어 있지 않기 때문에 영향을 받지 않는다.
어느 날 1번 컴퓨터가 웜 바이러스에 걸렸다. 컴퓨터의 수와 네트워크 상에서 서로 연결되어 있는 정보가 주어질 때, 1번 컴퓨터를 통해 웜 바이러스에 걸리게 되는 컴퓨터의 수를 출력하는 프로그램을 작성하시오.
입력
첫째 줄에는 컴퓨터의 수가 주어진다. 컴퓨터의 수는 100 이하이고 각 컴퓨터에는 1번 부터 차례대로 번호가 매겨진다. 둘째 줄에는 네트워크 상에서 직접 연결되어 있는 컴퓨터 쌍의 수가 주어진다. 이어서 그 수만큼 한 줄에 한 쌍씩 네트워크 상에서 직접 연결되어 있는 컴퓨터의 번호 쌍이 주어진다.
출력
1번 컴퓨터가 웜 바이러스에 걸렸을 때, 1번 컴퓨터를 통해 웜 바이러스에 걸리게 되는 컴퓨터의 수를 첫째 줄에 출력한다.
풀이 방법
BFS를 이용해서 풀었다. 1에서 시작해서 연결이 끊어지는 순간까지만 탐색하고, 그 순간까지 방문한 노드의 개수를 출력하면된다.
방문한 노드의 개수를 계산하는 방법은 방문할 때마다 해당 인덱스에 1을 넣어주고 최종적으로 합계를 구하는 방법도 있고, 방문할 때마다 +1씩을 더해준 다음 마지막에 최댓값을 구는 방법이 있다.
코드
# 1번에서 시작해 연결이 끊어질 때까지 노드를 탐색하고,
# 노드를 방문하면서 방문 횟수를 +1 해서 방문 여부 열에 넣어준다.
# 연결이 끊어지는 순간의 노드에 적힌 방문횟수를 출력한다.
from collections import deque
def bfs():
queue = deque()
queue.append(1)
visited[1] = 1
while queue:
node = queue.popleft()
for i in sorted(graph[node]):
if visited[i] == 0:
visited[i] = max(visited) + 1
queue.append(i)
return max(visited)-1
computer = int(input())
connection = int(input())
graph = [[] for _ in range(computer+1)]
visited = [0]*(computer+1)
for _ in range(connection):
a, b = map(int, input().split())
graph[a].append(b)
graph[b].append(a)
print(bfs())