[Algorithm] 프로그래머스 - 간단한 식 계산하기 (Kotlin)

문자열 binomial이 매개변수로 주어진다. binomial은 "a op b" 형태의 이항식이고 a, b는 음이 아닌 정수, op는 + / - / * 중 하나다. 주어진 식을 계산한 정수를 리턴하는 함수를 작성하라

 

'

 

a, op, b 사이에 공백이 있으니 split()으로 공백을 제거한다. 이후 만들어진 리스트의 1번 인덱스에는 항상 수식 기호가 들어가 있으니 이 기호가 무엇이냐에 따라 적절하게 리스트 안의 값들을 가져와서 계산하면 된다.

 

class Solution {
    fun solution(binomial: String): Int {
        val split = binomial.split(" ")

        return when (split[1]) {
            "+" -> split[0].toInt() + split[2].toInt()
            "-" -> split[0].toInt() - split[2].toInt()
            "*" -> split[0].toInt() * split[2].toInt()
            else -> 0
        }
    }
}