1987번: 알파벳 - Kotlin (DFS)

1987번: 알파벳

문제

세로 R칸, 가로 C칸으로 된 표 모양의 보드가 있다. 보드의 각 칸에는 대문자 알파벳이 하나씩 적혀 있고, 좌측 상단 칸 (1행 1열) 에는 말이 놓여 있다.

말은 상하좌우로 인접한 네 칸 중의 한 칸으로 이동할 수 있는데, 새로 이동한 칸에 적혀 있는 알파벳은 지금까지 지나온 모든 칸에 적혀 있는 알파벳과는 달라야 한다. 즉, 같은 알파벳이 적힌 칸을 두 번 지날 수 없다.

좌측 상단에서 시작해서, 말이 최대한 몇 칸을 지날 수 있는지를 구하는 프로그램을 작성하시오. 말이 지나는 칸은 좌측 상단의 칸도 포함된다.

입력

첫째 줄에 R과 C가 빈칸을 사이에 두고 주어진다. (1 ≤ R,C ≤ 20) 둘째 줄부터 R개의 줄에 걸쳐서 보드에 적혀 있는 C개의 대문자 알파벳들이 빈칸 없이 주어진다.

출력

첫째 줄에 말이 지날 수 있는 최대의 칸 수를 출력한다.

 

나의 풀이

 이 문제는 간단한 그래프 탐색 문제이다. 다만 같은 알파벳이 적힌 칸을 두 번 지날 수 없음으로 각 알파벳의 값으로 visited를 체크하여 같은 알파벳 나오지 않도록 했다.

import java.util.*
import kotlin.collections.ArrayList
import kotlin.math.max

val dx = arrayOf(0, 0, 1, -1)
val dy = arrayOf(1, -1, 0, 0)

lateinit var board : Array<CharArray>
var visited = Array(26) { false }
var result = 0

fun main() = with(System.`in`.bufferedReader()) {
    val (r, c) = readLine().split(" ").map{ it.toInt() }
    board = Array(r) { readLine().toCharArray() }

    visited[board[0][0] - 'A'] = true
    move(0, 0, 0, r, c)
    println(result + 1)
}

fun move(x: Int, y: Int, cnt: Int, r: Int, c: Int) {
    for (i in 0 until 4) {
        val tempX = x + dx[i]
        val tempY = y + dy[i]

        if (tempX in 0 until r &&
            tempY in 0 until c &&
            !visited[board[tempX][tempY] - 'A']) {
            visited[board[tempX][tempY] - 'A'] = true
            move(tempX, tempY, cnt + 1, r, c)
            result = max(result, cnt + 1)
            visited[board[tempX][tempY] - 'A'] = false
        }
    }
}