[Java] 문자열 계산하기

문제

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) {
        String[] oper = my_string.split(" ");
        int answer = Integer.parseInt(oper[0]);
        
        for(int i = 1; i < oper.length; i += 2) {
            if(oper[i].equals("+")) {
                answer += Integer.parseInt(oper[i+1]);
            }
            else {
                answer -= Integer.parseInt(oper[i+1]);
            }
        }
        return answer;
    }
}

주어진 문자열을 공백을 기준으로 split 한 값을 oper 배열에 담았다.

그리고 1번 인덱스부터 2칸씩 이동하면 + 또는 - 문자만 조회할 수 있고 문자열의 끝에는 항상 숫자로 끝나기 때문에 항상 i+1 번째 인덱스의 값을 더해나가면 해결할 수 있다.

다행히 연산자 우선순위가 없어서 쉬운 문제였다.

 

 

출처

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