[백준 2606번] 바이러스 (파이썬/python)

https://www.acmicpc.net/problem/2606

2606번: 바이러스

 

기초적인 bfs/dfs 문제. 둘 중 뭘 써도 상관 없는데

나는 파이썬 큐 자료구조를 이용해보고싶어서 bfs로 풀었다.

 

 

import sys
from collections import deque
 
input = sys.stdin.readline
 
n, t = int(input()), int(input())
li = [[0] * (n + 1) for _ in range(n + 1)]
visit = [0 for _ in range(n + 1)]
cnt = 0
 
 
def bfs():
    global cnt
    q = deque()
    q.append(1)
    visit[1] = 1
    while q:
        l = q.popleft()
 
        for i in range(1, n + 1):
            if li[l][i] == 1 and visit[i] == 0:
                visit[i] = 1
                cnt += 1
                q.append(i)
 
 
for _ in range(t):
    x, y = map(int, input().split())
    li[x][y] = 1
    li[y][x] = 1
 
bfs()
print(cnt)