1 분 소요

문제 설명

프로그래머스 181865번 간단한 식 계산하기

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

제한 사항

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

입출력 예시

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

입출력 예 설명 1

  • 예제 1번의 binomial은 “43 + 12”로 이 식을 계산한 결과인 43 + 12 = 55를 return 합니다.

입출력 예 설명 2

  • 예제 2번의 binomial은 “0 - 7777”로 이 식을 계산한 결과인 0 - 7777 = -7777을 return 합니다.

입출력 예 설명 3

  • 예제 3번의 binomial은 “40000 * 40000”으로 이 식을 계산한 결과인 40000 × 40000 = 1600000000을 return 합니다.

코드 구현

Python

def solution(binomial):
    binomial = binomial.split(" ")

    if binomial[1] == "+":
        return int(binomial[0]) + int(binomial[2])
    elif binomial[1] == "-":
        return int(binomial[0]) - int(binomial[2])
    elif binomial[1] == "*":
        return int(binomial[0]) * int(binomial[2])

Java

class Solution {
    public int solution(String binomial) {
        String[] split = binomial.split(" ");
        
        if (split[1].equals("+")) {
            return Integer.parseInt(split[0]) + Integer.parseInt(split[2]);
        } else if (split[1].equals("-")) {
            return Integer.parseInt(split[0]) - Integer.parseInt(split[2]);
        } else {
            return Integer.parseInt(split[0]) * Integer.parseInt(split[2]);
        }
    }
}