Programmers Level 0 - 문자열 계산하기

문제 설명

my_string은 "3 + 5"처럼 문자열로 된 수식입니다. 문자열 my_string이 매개변수로 주어질 때, 수식을 계산한 값을 return 하는 solution 함수를 완성해주세요.

 

제한사항

• 연산자는 +, -만 존재합니다.
• 문자열의 시작과 끝에는 공백이 없습니다.
• 0으로 시작하는 숫자는 주어지지 않습니다.
• 잘못된 수식은 주어지지 않습니다.
• 5 ≤ my_string의 길이 ≤ 100
• my_string을 계산한 결과값은 1 이상 100,000 이하입니다.
  • my_string의 중간 계산 값은 -100,000 이상 100,000 이하입니다.
  • 계산에 사용하는 숫자는 1 이상 20,000 이하인 자연수입니다.
  • my_string에는 연산자가 적어도 하나 포함되어 있습니다.
• return type 은 정수형입니다.
• my_string의 숫자와 연산자는 공백 하나로 구분되어 있습니다.

 

입출력 예

my_string	result
"3 + 4"	7

입출력 예 설명

입출력 예 #1

• 3 + 4 = 7을 return 합니다.

 

class Solution {
    public int solution(String my_string) {
        int answer = 0;
        String[] ar = my_string.split(" ");
        answer = Integer.parseInt(ar[0]);
        for(int i=1;i<ar.length-1;i+=2) {
            switch (ar[i]) {
            case "+":
                answer += Integer.parseInt(ar[i+1]);
                break;
            case "-":
                answer -= Integer.parseInt(ar[i+1]);
                break;
            }
        }
        return answer;
    }
}

 

공백을 구분자로 배열로 만든다.   연산자이면 다음 값을 읽어 연산자에 따라 계산해 준다.

 

class Solution {
    public int solution(String my_string) {
        return Arrays.stream(my_string.replaceAll("- ", "-").replaceAll("[+] ", "").split(" "))
                     .mapToInt(Integer::parseInt)
                     .sum();
    }
}

 

+공백은 지우고 -공백은 -로 변경하고 공백으로 구분하여 배열을 만들면 "3 + 4 - 7 + 3"은 [3,4,-7,3]이 된다. 그러면 Stream으로 만들어 모두 더하면 된다.