[Kotlin] 백준 16506번 CPU

출처: unsplash.com

 

CPU(16506번)

Silver 5

#구현

 

문제 내용

 

 

문제 접근

위 표를 참고해서 opcode, rD, rA, rB(#C)가 주어질 때, 문제에 정의된 번역 방법으로 어셈블리어를 번역해서 기계어로 출력한다.

각 문제에 필요한 조건들이 뭐가 있는지 정리해 보겠다. 기계어는 총 16개의 0과 1로 이루어져 있다.

16개의 수들이 적용되는 조건은 다음과 같다.

 

1. 0~3번째까지의 수는 opcode에 맞게 구성해 주면 된다. 위 표 참고.

2. 4번째 수는 opcode가 C로 끝나면 1, 아니면 0

3. 5번째 수는 0 고정.

4. 6~8번째 수는 rD의 2진수값.

5. 9~11번째 수는 rA의 2진수값, NOT의 경우는 0 고정.

6.12~15번째 수는 opcode가 C로 끝날 때는 #C의 2진수값, 아니면 rB의 2진수값과 15번째 수는 0 고정.

 

아래의 조건들을 구현해 주면 된다.

 

<조건 1>

우선 1번 조건의 경우는 25가지의 수에서 2개씩 묶여있기 때문에(NOT 제외) contains를 이용해 2가지의 입력 경우를 한 번에 처리할 수 있다.

if (opCode.contains("ADD"))

ADD를 포함하는 조건을 적용하면 ADD와 ADDC 모두를 저격할 수 있다. 어차피 ADD와 ADDC의 0~3자리 수를 구할 수 있다.

 

<조건 2>

4번째 수를 구하기 위해서는 opcode의 마지막 수가 C로 끝나는지 확인한다.

if (opCode.last() == 'C')

5번째 수는 0 고정이기 때문에 같이 넣어줄 수 있다.

 

<조건 3>

6~8번째 수를 구하는 건 딱히 조건이 필요하지 않아서 넘어가고, 9~11번째 수를 구하기 위해서는 opcode가 MOV, MOVC 거나 NOT인 경우만 000이고 나머지는 rA의 2진수값이다.

if (opCode.contains("MOV") || opCode.contains("NOT"))

 

<조건 4>

12~15번째 수는 좀 복잡할 수 있다.

마지막 opcode가 C로 끝날 때는 rB를 4자리의 2진수로 나타내고, 아닌 경우는 3자리의 2진수와 끝에 0을 붙여준다.

if (opCode.last() == 'C')

 

마지막으로 진수 변환 결과를 3자리 또는 4자리로 나타내야 한다.

그것을 나타내기 위해서는 string으로 변환 후, padStart()를 이용해 문자열의 length를 고정하고, length가 부족한 경우 특정 문자로 채울 수 있다.

rB.toString(2).padStart(4, '0')

위 코드는 length를 4로 설정하고, 부족한 부분은 앞을 0으로 채운다.

예를 들어 11이 들어오면 0011이 되게 된다.

 

 

문제 해결 코드

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

fun main() = with(BufferedReader(InputStreamReader(System.`in`))) {
    val case = readLine().toInt()

    for (i in 0 until case) {
        val assembly = readLine().split(" ")
        val opCode = assembly[0]
        val (rD, rA, rB) = assembly.subList(1, assembly.size).map { it.toInt() }
        val machineLanguage = StringBuilder()

        // 0 ~ 3번
        if (opCode.contains("ADD")) machineLanguage.append("0000")
        else if (opCode.contains("SUB")) machineLanguage.append("0001")
        else if (opCode.contains("MOV")) machineLanguage.append("0010")
        else if (opCode.contains("AND")) machineLanguage.append("0011")
        else if (opCode.contains("OR")) machineLanguage.append("0100")
        else if (opCode.contains("NOT")) machineLanguage.append("0101")
        else if (opCode.contains("MULT")) machineLanguage.append("0110")
        else if (opCode.contains("LSFTL")) machineLanguage.append("0111")
        else if (opCode.contains("LSFTR")) machineLanguage.append("1000")
        else if (opCode.contains("ASFTR")) machineLanguage.append("1001")
        else if (opCode.contains("RL")) machineLanguage.append("1010")
        else if (opCode.contains("RR")) machineLanguage.append("1011")

        // 4번
        if (opCode.last() == 'C') machineLanguage.append("1")
        else machineLanguage.append("0")

        // 5번
        machineLanguage.append("0")

        // 6 ~ 8번
        machineLanguage.append(rD.toString(2).padStart(3, '0'))

        // 9 ~ 11번
        if (opCode.contains("MOV") || opCode.contains("NOT")) machineLanguage.append("000")
        else machineLanguage.append(rA.toString(2).padStart(3, '0'))

        // 12 ~ 15번
        if (opCode.last() == 'C') {
            machineLanguage.append(rB.toString(2).padStart(4, '0'))
        } else {
            machineLanguage.append("${rB.toString(2).padStart(3, '0')}0")
        }

        println(machineLanguage.toString())
    }
}

 

문제 풀이

입력을 받고 조건 1, 2, 3, 4를 적용해 16자리의 기계어로 번역해서 출력하면 된다.

 

 

문제 해결 과정

구현해야 하는 조건이 많다. 또한 padStart 구현 위치를 잘못 잡아서 헤맸었던 적도 있다.