문자열 계산하기

/* 문자열 계산하기

 * my_string은 "3 + 5"처럼 문자열로 된 수식입니다.

 * 문자열 my_string이 매개변수로 주어질 때,

 * 수식을 계산한 값을 return 하는 solution 함수를 완성해주세요.

 *

 * my_string    result

 * "3 + 4"      7       3 + 4 = 7을 return 합니다.

 */

 

public class programmer_0_17 {
    static String a1 = "3 + 4";
 
    public int cal(String n, int input, int rst){
        switch(n){
            case "+" : 
            rst = rst + input;
                break;
            case "-" : 
            rst = rst - input;
                break;     
        } 
        return rst;
    }
    public int solution(String my_string) {
        int answer = 0;
        String[] arg = my_string.split(" ");
 
        for(int i = 0; i < arg.length; i++){
            if(i == 0) answer = Integer.parseInt(arg[i]);
            else if(i%2 == 0){
                answer = cal(arg[i-1], Integer.parseInt(arg[i]), answer);
            }
        }
        return answer;
    }
    public static void main(String args[]){
        programmer_0_17 t = new programmer_0_17();
        System.out.println("---------------------------------------");
        System.out.println("result = " + t.solution(a1));
        System.out.println("---------------------------------------");
    }
}