[프로그래머스] 네트워크 | kotlin

문제 설명

네트워크란 컴퓨터 상호 간에 정보를 교환할 수 있도록 연결된 형태를 의미합니다. 예를 들어, 컴퓨터 A와 컴퓨터 B가 직접적으로 연결되어있고, 컴퓨터 B와 컴퓨터 C가 직접적으로 연결되어 있을 때 컴퓨터 A와 컴퓨터 C도 간접적으로 연결되어 정보를 교환할 수 있습니다. 따라서 컴퓨터 A, B, C는 모두 같은 네트워크 상에 있다고 할 수 있습니다.

컴퓨터의 개수 n, 연결에 대한 정보가 담긴 2차원 배열 computers가 매개변수로 주어질 때, 네트워크의 개수를 return 하도록 solution 함수를 작성하시오.

---

풀이

Dfs를 이용하여 풀어봤다.

class Solution {
    private val visit: MutableList<Int> = mutableListOf()

    fun solution(n: Int, computers: Array<IntArray>): Int {
        var answer = 0

        for (i in computers.indices) {
            if (visit.contains(i)) continue
            checkVisit(computers, i)
            answer++
        }
        return answer
    }

    private fun checkVisit(computers: Array<IntArray>, index: Int) {
        computers[index].forEachIndexed { i, it ->
            if (it == 1 && !visit.contains(i)) {
                visit.add(i)
                checkVisit(computers, i)
            }
        }
    }
}

사실 Visit에는 boolean 타입을 넣어서 방문여부를 확인했던걸로 아는데 문제 풀 당시 잘 기억이 나질 않아서 그냥 값을 넣어버렸다.. 

 

사실 알고리즘 배울 때 개념은 이해하고 코드로 구현하면서부터 이해가 잘 안가서 힘들어했는데 문제로 접하니 어찌저찌 적용이 되네...

 

추후에 다른 방법으로 풀이해 볼 예정!

 

문제 링크