[프로그래머스] 간단한 식 계산하기 --java

https://school.programmers.co.kr/learn/courses/30/lessons/181865

프로그래머스

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

 

 

 

 

 

 

 

풀이)

class Solution {
    public int solution(String binomial) {
        int answer = 0;
        int idx = 0;
        
        binomial = binomial.replaceAll("\\s+", "");

        if(binomial.indexOf("+")!=-1){
            idx = binomial.indexOf("+");
            int a = Integer.parseInt(binomial.substring(0,idx));
            int b = Integer.parseInt(binomial.substring(idx+1, binomial.length()));
            answer = a+b;

        }else if(binomial.indexOf("-")!=-1){
            idx = binomial.indexOf("-");
            int a = Integer.parseInt(binomial.substring(0,idx));
            int b = Integer.parseInt(binomial.substring(idx+1, binomial.length()));
            answer = a-b;

        }else if(binomial.indexOf("*")!=-1){
            idx = binomial.indexOf("*");
            int a = Integer.parseInt(binomial.substring(0,idx));
            int b = Integer.parseInt(binomial.substring(idx+1, binomial.length()));
            answer = a*b;
        }
        
        return answer;
    }
}

나는 공백을 활용해서 풀생각을 못하고, 계산에 방해되니까 문자열안에 있는 공백부터 없앴다...ㅋㅋㅋ