[백준/코틀린] 2606번: 바이러스

링크

2606번: 바이러스

풀이

1번 컴퓨터에서 시작하여 연결된 모든 컴퓨터를 탐색합니다.
도달할 수 있는 모든 노드를 탐색하기 위해 DFS 또는 BFS를 사용합니다.

코드

lateinit var graph: List<MutableList<Int>>
lateinit var visited: BooleanArray
val stack = ArrayDeque<Int>()
var ans = -1

fun dfs(s: Int) {
    visited[s] = true
    stack.addLast(s)
    while (stack.isNotEmpty()) {
        val cur = stack.removeLast()

        ans++
        graph[cur].forEach {
            if (!visited[it]) {
                visited[it] = true
                stack.addLast(it)
            }
        }
    }
}

fun main() {
    val n = readln().toInt()
    graph = List(n + 1) { mutableListOf() }
    visited = BooleanArray(n + 1)

    repeat(readln().toInt()) {
        val (u, v) = readln().split(" ").map { it.toInt() }

        graph[u] += v
        graph[v] += u
    }

    dfs(1)

    println(ans)
}