[프로그래머스_Java] Lv.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의 숫자와 연산자는 공백 하나로 구분되어 있습니다.

풀이

class Solution {
    public int solution(String my_string) {
        int answer = 0;
        String[] str = my_string.split(" "); // 1
        answer += Integer.parseInt(str[0]); // 2

        for (int i=0; i<str.length-1; i++) { // 3
             if (str[i].equals("+")) {
                answer += Integer.parseInt(str[i+1]);
            } else if (str[i].equals("-")) {
                answer -= Integer.parseInt(str[i+1]);
            }
        }
       
        return answer;
    }
}

1. 공백을 기준으로 문자열을 분리하여 배열에 저장
2. 첫 숫자를 answer에 더해줌
3. 배열을 순회하며 수식이 "+"인 경우에는 다음 숫자를 더하고, "-"인 경우에는 빼줌

 

문제를 풀 때마다 이게 최선일까 하는 고민이 생긴다
분명 더 효율적인 풀이가 있을 것 같은데....
더 깊은 공부가 필요할 듯하다

 

프로그래머스