알고리즘 문제 풀이 - 문자열 계산하기

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

이런 문제인데

public int solution(String my_string) {
    int answer = 0;
    String[] arrMy_string = my_string.split(" ");
    int num1 = Integer.parseInt(arrMy_string[0]);
    int num2 = Integer.parseInt(arrMy_string[2]);

    if (arrMy_string[1].equals("+")) {
        answer = num1 + num2;
    } else {
        answer = num1 - num2;
    }

    return answer;
}

이렇게 접근해보았다.

입출력 예시에 있는 "3 + 4"는 정답으로 나왔지만
제출 후 채점하기를 해보니 테스트 10개 중 2개만 정답, 8개가 오답으로 나왔다.

제한사항을 다시 읽어봐도 이유를 알 수가 없어서 구글링을 해보니
연산자가 사용되는 수식이 하나가 아닐 수 있다는 글이 있었다.

https://school.programmers.co.kr/questions/38313

하핳... 저렇게는 생각을 안해봤는데
입출력 예시에 연산자 2개짜리로 하나만 더 알려주지...

어쩐지 제한사항에
'my_string의 중간 계산 값은 -100,000 이상 100,000 이하입니다.'
이걸보고 중간 계산이 뭐지..? 싶었는데
'my_string에는 연산자가 적어도 하나 포함되어 있습니다.'
여기서 '적어도'라는 힌트도 있었다.

결국은 for문을 사용하기로 했다.

public int solution(String my_string) {
    String[] arrMy_string = my_string.split(" ");
    int answer = Integer.parseInt(arrMy_string[0]);

    for (int i = 1; i < arrMy_string.length; i += 2) {
        if (arrMy_string[i].equals("+")) {
            answer += Integer.parseInt(arrMy_string[i + 1]);
        } else {
            answer -= Integer.parseInt(arrMy_string[i + 1]);
        }
    }

    return answer;
}