[Kotlin, S4] 백준 9242번 폭탄 해체

출처: unsplash.com

 

폭탄 해체(9242번)

Silver 4

#구현 #문자열 #파싱

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

 

문제 내용

 

 

문제 접근

별 기호(*)와 공백으로 이루어진 숫자가 입력된다.

숫자는 7 segment 숫자 표기법으로 이루어져 있다.

그냥 숫자를 파싱 해서 6으로 나누어 떨어지면 beer를 출력하고 그렇지 않으면 boom을 출력하면 되는 간단한 문제이다.

 

또한 암호가 무조건 7 segment의 숫자들만 나오는 것도 아니다.

암호는 3 * 5의 형태로 1칸씩 공백을 띄고 주어지게 되는데, 3 * 5의 암호 형식을 지키지 않은 코드가 올 수 도 있다.

그래서 위 2가지 상황에 맞는 처리(boom)를 해주어야 한다.

 

문제 해결 코드

import java.io.BufferedReader
import java.io.InputStreamReader

fun main() = with(BufferedReader(InputStreamReader(System.`in`))) {
    val codes = mutableListOf<List<String>>()
    for (i in 0 until 5) {
        val codeLine = readLine().map { it.toString() }
        codes.add(codeLine)
    }

    try {
        var codeParseResult = ""
        val subCode = mutableListOf<List<String>>()
        for (j in codes[0].indices) {
            if (j % 4 == 0) {
                for (k in 0 until 5) {
                    val subCodeLine = codes[k].subList(j, j + 3)
                    subCode.add(subCodeLine)
                }

                val res = parseCode(subCode)
                codeParseResult += res
                subCode.clear()
            }
        }

        if (codeParseResult.toInt() % 6 == 0) println("BEER!!")
        else println("BOOM!!")
    } catch (e: Exception) {
        println("BOOM!!")
    }
}

private fun parseCode(compareValue: List<List<String>>): String {
    val zero = listOf(
        listOf("*", "*", "*"),
        listOf("*", " ", "*"),
        listOf("*", " ", "*"),
        listOf("*", " ", "*"),
        listOf("*", "*", "*"),
    )
    val one = listOf(
        listOf(" ", " ", "*"),
        listOf(" ", " ", "*"),
        listOf(" ", " ", "*"),
        listOf(" ", " ", "*"),
        listOf(" ", " ", "*")
    )
    val two = listOf(
        listOf("*", "*", "*"),
        listOf(" ", " ", "*"),
        listOf("*", "*", "*"),
        listOf("*", " ", " "),
        listOf("*", "*", "*"),
    )
    val three = listOf(
        listOf("*", "*", "*"),
        listOf(" ", " ", "*"),
        listOf("*", "*", "*"),
        listOf(" ", " ", "*"),
        listOf("*", "*", "*"),
    )
    val four = listOf(
        listOf("*", " ", "*"),
        listOf("*", " ", "*"),
        listOf("*", "*", "*"),
        listOf(" ", " ", "*"),
        listOf(" ", " ", "*")
    )
    val five = listOf(
        listOf("*", "*", "*"),
        listOf("*", " ", " "),
        listOf("*", "*", "*"),
        listOf(" ", " ", "*"),
        listOf("*", "*", "*"),
    )
    val six = listOf(
        listOf("*", "*", "*"),
        listOf("*", " ", " "),
        listOf("*", "*", "*"),
        listOf("*", " ", "*"),
        listOf("*", "*", "*")
    )
    val seven = listOf(
        listOf("*", "*", "*"),
        listOf(" ", " ", "*"),
        listOf(" ", " ", "*"),
        listOf(" ", " ", "*"),
        listOf(" ", " ", "*")
    )
    val eight = listOf(
        listOf("*", "*", "*"),
        listOf("*", " ", "*"),
        listOf("*", "*", "*"),
        listOf("*", " ", "*"),
        listOf("*", "*", "*")
    )
    val nine = listOf(
        listOf("*", "*", "*"),
        listOf("*", " ", "*"),
        listOf("*", "*", "*"),
        listOf(" ", " ", "*"),
        listOf("*", "*", "*")
    )

    return when (compareValue) {
        zero -> "0"
        one -> "1"
        two -> "2"
        three -> "3"
        four -> "4"
        five -> "5"
        six -> "6"
        seven -> "7"
        eight -> "8"
        nine -> "9"
        else -> throw Exception("올바르지 않은 숫자 형식")
    }
}

 

문제 풀이

코드 형식이 한 줄에 3개의 문자 또는 공백과 1개의 공백이 반복되는 형식이기 때문에, 코드의 단어의 첫 번째 차례가 되면 반복으로 입력받은 코드의 3개의 단어를 5번 반복한다.

예를 들어 입력이 아래와 같이 주어질 때,

***   * * * *** *** *** ***
* *   * * *   *   *   * *  
* *   * *** *** *** *** ***
* *   *   * *     * *   * *
***   *   * *** *** *** ***

아래와 같이 코드 단어를 1개씩 불러오는 방식이다.

 

***
* *
* *
* *
***

  *
  *
  *
  *
  *
  
* *
* *
***
  *
  *
  
. . .

아래와 같이 단어를 1개씩 불러온다.

그리고 불러온 단어를 2중 리스트로 저장하고, parseCode를 이용해 숫자로 변환한다.

 

parseCode에는 0부터 9까지의 7 segment 숫자 표기법 형태로 정의한 후, 들어온 code와 같은 수가 있는지 확인한다.

0부터 9까지의 수가 모두 해당이 안 되면 올바르지 않은 코드가 온 것이다.

그래서 Exception을 throw 해주고 parseCode 로직을 try catch에서 진행한다.

catch로 온 경우에는 parseCode에서 error가 던져진 상황에 오기 때문에 boom을 출력하고, 제대로 된 숫자를 파싱 한 후, int로 바꾼 수가 6으로 나누어 떨어진 경우는 beer를, 그렇지 않으면 boom을 출력한다.

 

 

문제 해결 과정

단순 노가다 구현 문제이다.

0부터 9까지의 7 segment 숫자 표기법을 미리 정의하고 올바른 수를 parsing 해야 한다.

하지만 올바르지 않은 경우에 대한 처리를 해야 한다. 이는 근데 많이 어렵지 않다.

코드가 올바르지 않은 경우에 대한 처리를 빼먹어 틀리는 일이 없도록 하자.