[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의 숫자와 연산자는 공백 하나로 구분되어 있습니다.

입출력 예

입출력 예 설명

입출력 예 #1

• 3 + 4 = 7을 return 합니다.

코드 설명

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Solution {
	public int solution(String my_string) {
        int answer = 0;
        String[] arr = my_string.split(" "); // arr에 띄어쓰기 제거한 my_string을 저장
        answer = Integer.parseInt(arr[0]); // answer에 0번째 값을 미리 저장해줌
        
        for (int i = 1; i < arr.length; i++) { // 1번째 값부터 arr의 length 전 까지 
        	if(i%2!=0) { // 만약 i번째가 1,3,5 홀수 자리일 때(이유는 홀수 자리가 +,- 로 이루어져 있으므로)
        		if (arr[i].equals("+")){ // arr[i]가 + 가 들어오면 
                    answer += Integer.parseInt(arr[i+1]); // answer에 + 자리 다음 값을 더해줌  +자리 다음 값은 더해야 할 숫자 이므로
                } 
            	else {
                    answer -= Integer.parseInt(arr[i+1]); // answer에 - 자리 다음 값을 빼줌    -자리 다음 값은 빼야 할 숫자 이므로
                } 
        	}   
        }
 // 이런식으로 푼 이유는 1+2 가 들어올 수 있지만 1+2+3+4 여러 수식이 들어올 수 있기 때문에.
        
        return answer;
    }
	public static void main(String[] args) throws IOException {
		Solution T = new Solution();
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine(), "\n");
		String my_string = st.nextToken();
		System.out.println(T.solution(my_string));
		br.close();
	}
}