[Lv. 0] 간단한 식 계산하기

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/181865

프로그래머스

난이도 : Level.0

언어 : Kotlin

 

문제 설명 :

문자열 binomial이 매개변수로 주어집니다. binomial은 "a op b" 형태의 이항식이고 a와 b는 음이 아닌 정수, op는 '+', '-', '*' 중 하나입니다. 주어진 식을 계산한 정수를 return 하는 solution 함수를 작성해 주세요.

 

제한 사항 : 

• 0 ≤ a, b ≤ 40,000
• 0을 제외하고 a, b는 0으로 시작하지 않습니다.

입출력 예 :

binomial	result
"43 + 12"	55
"0 - 7777"	-7777
"40000 * 40000"	1600000000

풀이 : 

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

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