백준 13023 ABCDE Kotlin (dfs)

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

13023번: ABCDE

문제

BOJ 알고리즘 캠프에는 총 N명이 참가하고 있다. 사람들은 0번부터 N-1번으로 번호가 매겨져 있고, 일부 사람들은 친구이다.

오늘은 다음과 같은 친구 관계를 가진 사람 A, B, C, D, E가 존재하는지 구해보려고 한다.

• A는 B와 친구다.
• B는 C와 친구다.
• C는 D와 친구다.
• D는 E와 친구다.

위와 같은 친구 관계가 존재하는지 안하는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 사람의 수 N (5 ≤ N ≤ 2000)과 친구 관계의 수 M (1 ≤ M ≤ 2000)이 주어진다.

둘째 줄부터 M개의 줄에는 정수 a와 b가 주어지며, a와 b가 친구라는 뜻이다. (0 ≤ a, b ≤ N-1, a ≠ b) 같은 친구 관계가 두 번 이상 주어지는 경우는 없다.

출력

문제의 조건에 맞는 A, B, C, D, E가 존재하면 1을 없으면 0을 출력한다.

알고리즘 분류

• 그래프 이론
• 그래프 탐색
• 깊이 우선 탐색

풀이

간단한 dfs 문제이다.

핵심은 dfs를 돌리는데 depth가 4인 경우가 있는지 없는지를 찾는 것이다.

위의 예시를 보면 A->B->C->D->E

A의 depth를 0이라 하면, 이전에 방문한 노드를 제외하고 탐색하여 E까지 도달했고, depth가 4가 된다.

현재 시작점은 A로 했지만, 탐색할 땐 A부터 E까지 모두 시작점으로 dfs를 돌려보아야 한다.

코드

val br = System.`in`.bufferedReader()

lateinit var edge: Array<ArrayList<Int>>

var result = false

fun dfs(cur: Int, depth: Int, visited: BooleanArray) {

    if(depth==4){
        result =true
        return
    }
    for(next in edge[cur]){
        if(visited[next]) continue
        visited[next] = true
        dfs(next, depth+1, visited)
        visited[next] = false
    }

}

fun main() = with(System.out.bufferedWriter()) {

    //input
    val (n, m) = br.readLine().split(' ').map { it.toInt() }
    edge = Array(n) { ArrayList() }
    repeat(m) {
        val (a, b) = br.readLine().split(' ').map { it.toInt() }
        edge[a].add(b)
        edge[b].add(a)
    }

    //solve,output
    for (i in 0 until n) {
        dfs(i, 0, BooleanArray(n) { it == i })
        if (result) {
            write("1")
            close()
            return
        }
        result = false
    }

    write("0")
    close()
}