[백준] 2606 (바이러스)

#1 알고리즘

1

 

#2 코드 - 코틀린

fun main() {
    readln() // 컴퓨터의 갯수는 풀이에 사용하지 않는다.
    val connectionCount = readln().toInt()

    val connections : ArrayList<Array<Int>> = ArrayList()
    for(i : Int in 0..<connectionCount) {
        val digitChars = readln().split(" ")
        connections.add( arrayOf(digitChars[0].toInt(), digitChars[1].toInt()) )
    }

    val virusHosts : ArrayList<Int> = ArrayList()

    recursiveFindNextHost(connections, 1, virusHosts)

    println(virusHosts.size - 1) // 1번 컴퓨터 제외
}

fun recursiveFindNextHost(connections : ArrayList<Array<Int>>, computerNumber : Int, virusHosts : ArrayList<Int>) {
    virusHosts.addIfNotExist(computerNumber)

    for(i : Int in 0..<connections.size) {
        if(connections[i].contains(computerNumber)) {

            val anotherComputerNumber = if(connections[i][0] == computerNumber) { connections[i][1] } else { connections[i][0] }
            connections[i] = arrayOf(-1, -1) // connections[i]는 정보를 취했으니 이제 쓸모가 없다. 따라서, -1를 채워넣는다.

            recursiveFindNextHost(connections, anotherComputerNumber, virusHosts)
        }
    }
}

fun <T> ArrayList<T>.addIfNotExist(value: T) {
    if(!this.contains(value)) {
        this.add(value)
    }
}

가독성을 향상을 위해 확장함수를 사용했다.