본문 바로가기

알고리즘/DFS | BFS

파이썬 | 백준 | 2606 | 바이러스

solution

1. bfs : 84ms

2. dfs 이용

# 2606, 바이러스
import sys
input = sys.stdin.readline


def dfs(x):
    visited[x] = 1
    global virus

    for i in network[x]:
        if not visited[i]:
            virus += 1
            dfs(i)
    return virus


n = int(input())
m = int(input())
network = [[] for _ in range(n+1)]
for _ in range(m):
    a, b = map(int, input().split())
    network[a].append(b)
    network[b].append(a)
visited = [0]*(n+1)
virus = 0
print(dfs(1))

29380KB

68ms

 

문제 출처 https://www.acmicpc.net/problem/2606

 

2606번: 바이러스

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

www.acmicpc.net