[백준/BOJ] 11724번: 연결 요소의 개수

https://www.acmicpc.net/problem/11724

11724번: 연결 요소의 개수

문제

방향 없는 그래프가 주어졌을 때, 연결 요소 (Connected Component)의 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 정점의 개수 N과 간선의 개수 M이 주어진다. (1 ≤ N ≤ 1,000, 0 ≤ M ≤ N×(N-1)/2) 둘째 줄부터 M개의 줄에 간선의 양 끝점 u와 v가 주어진다. (1 ≤ u, v ≤ N, u ≠ v) 같은 간선은 한 번만 주어진다

출력

첫째 줄에 연결 요소의 개수를 출력한다.

해설

dfs와 인접행렬을 이용하여 문제를 해결할 수 있다. 

예제 입력1을 예시로 들어보자.

방향성이 없는 그래프이므로 양방향으로 체크를 해주어야 한다.

arr[a][b] = true
arr[b][a] = true

체크를 다 끝냈다면

1부터 노드의 갯수만큼 dfs통해 탐색을 한다.

 

반복 중 아직 방문하지 않은 곳에 도착했다면 count를 증가시킨 후, 방문 체크를 해준다.

모든 곳을 방문한 후 결과는 다음과 같다.

두개의 사이클로 나눠지므로 답은 2가 된다.

해설

private lateinit var arr: Array<IntArray>
private lateinit var visited: BooleanArray

fun main() = with(System.`in`.bufferedReader()) {
    val (nodeCnt, edgeCnt) = readLine().split(" ").map { it.toInt() }
    arr =  Array(nodeCnt + 1) { IntArray(nodeCnt + 1) }
    visited = BooleanArray(nodeCnt + 1)

    repeat(edgeCnt) {
        val(start, end) = readLine().split(" ").map { it.toInt() }
        arr[start][end] = 1
        arr[end][start] = 1
    }

    var cnt = 0
    for(i in 1 .. nodeCnt) {
        if(!visited[i]) {
            cnt++
            dfs(i, nodeCnt)
        }
    }

    println(cnt)
}

private fun dfs(node: Int, nodeCnt: Int) {
    visited[node] = true

    for(i in 1 .. nodeCnt) {
        if(arr[node][i] == 1 && !visited[i]) dfs(i, nodeCnt)
    }
}