자바 알고리즘 - 문자열 계산하기

문제 설명

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[] arr = my_string.split(" ");
        int answer = 0;
        boolean chk = true;
        
        for(int i=0; i<arr.length; i++) {
            if(i%2==0) {
                answer+=chk?Integer.parseInt(arr[i]):-Integer.parseInt(arr[i]);
            }else {
                chk = arr[i].equals("+");
            }
        }
        return answer;
    }
}

먼저 split() 함수를 통해 my_string을 배열로 만들어준다. 그리고 부호 체크를 위한 boolean 타입 변수를 선언한다.

for문을 돌면서 i가 2로 나눴을 때 나머지가 0일 때, 즉 원소가 숫자일 때만 더해준다.

그 때 chk 변수를 확인하여 true면 양수를, false면 음수를 더한다.

chk 변수는 원소가 숫자가 아닐 때 "+" 기호와 비교하여 일치하면 true, 불일치하면 false를 담아두고 이후 answer에 더할 원소 부호를 결정한다.