[코틀린] 백준 3190번: 뱀 - 삼성 SW 역량 테스트 기출 문제 풀이

문제 풀러 가기!

풀이

구현 문제입니다만 신경써야할 부분들이 꽤 있던 문제였습니다.
문제에서 머리가 먼저 선진입 하고, 꼬리가 뒤따라 없어지는 로직으로 구성되었는데요, 게임 오버 조건을 진입 후에 넣어주는 것이 포인트입니다.

방향 전환 시점도 잘 고려해보시기바랍니다.

참고로 게임 오버 조건이 머리가 몸의 일부에 닿았을 때 성립이 되기 때문에 몸통에 해당하는 좌표를 찍어 주시는 것도 도움됩니다. 저 같은 경우에는 9로 나타내었네요.

package 삼성SW역량테스트.`3190_뱀`

import java.util.*

val dr = listOf(-1, 0, 1, 0) // 북 동 남 서
val dc = listOf(0, 1, 0, -1)
fun main() {
    val n = readln().toInt()
    val board = Array(n) {IntArray(n)}
    val k = readln().toInt()

    for (i in 0 until k) {
        val (r, c) = readln().split(" ").map { it.toInt() - 1 }

        board[r][c] = 1
    }
    val l = readln().toInt()
    val moveList: Queue<Command> = LinkedList()
    for (i in 0 until l) {
        val st = StringTokenizer(readln())

        val x = st.nextToken().toInt()
        val c = st.nextToken()

        moveList.add(Command(x, c))
    }
    val snake = ArrayDeque<Snake>()
    snake.add(Snake(0, 0)) // 첫 시작 위치 설정
    var dir = 1
    var count = 0
    while (true) {
        // 현재 방향으로 한 칸 이동
        val nr = snake.peekFirst().r + dr[dir]
        val nc = snake.peekFirst().c + dc[dir]
        snake.addFirst(Snake(nr, nc))

        // 벽과 부딪히면 게임은 끝남
        if (!(snake.peekFirst().r in board.indices && snake.peekFirst().c in board[0].indices)) break

        // 자기 자신의 몸과 부딪히면 게임은 끝남
        if (board[nr][nc] == 9) break

        // 사과가 없으면 꼬리를 자른다.
        if (board[nr][nc] != 1) {
            val last = snake.pollLast()
            board[last.r][last.c] = 0
        }
        // 사과는 먹어서 없어진다.
        board[nr][nc] = 9


        count++
        // 만약 방향 전환 명령이 있으면, 방향 전환을 한다.
        moveList.peek()?.let {
            val peek = moveList.peek()
            if (count == peek.x) {
                moveList.poll()
                when (peek.c) {
                    "L" -> {
                        // 왼쪽 방향 전환
                        dir = (dir + 3) % 4
                    }
                    "D" -> {
                        // 오른쪽 방향 전환
                        dir = (dir + 1) % 4
                    }
                }
            }
        }
    }
    println(++count)
}
fun Array<IntArray>.printArray() {
    for (i in indices) {
        println(get(i).contentToString())
    }
}
data class Command(val x: Int, val c: String)
data class Snake(val r: Int, val c: Int)