[프로그래머스] 문자열 계산하기

📌 문제 설명

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

📌 코드

import java.util.StringTokenizer;

class Solution {
    public int solution(String my_string) {
     StringTokenizer st = new StringTokenizer(my_string, " ");
        String [] element = new String [100];
        int index = 0, answer = 0;
        while(st.hasMoreTokens())
                element[index++] = st.nextToken();
        for(int i=0 ; i < index; i++)
        {
            if (i == 0)
                answer = Integer.parseInt(element[i]);

            if(element[i].equals("+"))
                answer += Integer.parseInt(element[i+1]);
            else if(element[i].equals("-"))
                answer -= Integer.parseInt(element[i+1]);
        }
        return answer;
    }
}

📌 문제 해결 과정

💡 while문과 StringTokenizer 사용하면서 토큰이 있을 때까지 배열에 집어넣어 주며 index로 몇 개의 토큰이 저장되었는지 확인할 수 있게 하기

💡 저장된 토큰 개수만큼 (index만큼) for문을 돌려 주는데. i가 0일 때는 배열 맨 첫 번째의 숫자를 다루게 되므로 answer이라는 변수에 첫 번째 숫자를 그대로 넣어 주기

💡 배열에 있는 원소와 연산자가 일치하면 answer에 있는 값에 연산자가 들어 있던 다음 원소 값을 answer에 더해 주거나 빼 주게 됨 > 맨 처음을 제외하고는 배열의 원소에 숫자가 들어 있을 때는 반복문이 바로 넘어가는 것과 같아짐

첫 포스팅인데 이렇게 하는 게 맞는 건지 잘 모르겠어요 일단 마크다운 더 알아보고 수정도 해야 될 것 같은 😂