코딩테스트/BOJ

[백준] [BFS] 2606. 바이러스

박소민 2023. 3. 5. 15:50
2606.바이러스
 

2606번: 바이러스

첫째 줄에는 컴퓨터의 수가 주어진다. 컴퓨터의 수는 100 이하이고 각 컴퓨터에는 1번 부터 차례대로 번호가 매겨진다. 둘째 줄에는 네트워크 상에서 직접 연결되어 있는 컴퓨터 쌍의 수가 주어

www.acmicpc.net

 

  • 내 풀이 - Python
#연결된 컴퓨터부터 전파
#큐 사용 -> bfs
from collections import deque
from collections import defaultdict

n=int(input())
m=int(input())
lst=defaultdict(list)
queue=deque()
visited=[False]*(n+1)
maxCnt=0
for _ in range(m):
    s,e=map(int,input().split())
    lst[s].append(e)
    lst[e].append(s)

def bfs(start):
    global maxCnt
    queue.append(start)
    visited[start]=True
  
    while queue:
        q= queue.popleft()
  
        for ls in lst[q]:
            if visited[ls]==False:
                visited[ls]=True
                maxCnt+=1
                queue.append(ls)

bfs(1)
print(maxCnt)

 

  • java
https://yygs321.tistory.com/283