SWEA_D4_1223

📁 [D4_1223] 계산기2

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


public class Solution {

	static StringTokenizer st;
	static String temp = "";

	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();

		for (int t = 1; t <= 10; t++) {
			sb.append("#").append(t).append(" ");

			// 계산식 길이
			int N = Integer.parseInt(br.readLine());

			// 문자열
			String str = br.readLine();

			Stack<String> stack = new Stack<String>();
			
			// 첫번째 수 입력
			stack.add(String.valueOf(str.charAt(0)));
			for (int i = 1; i < N; i++) {
				stack.add(String.valueOf(str.charAt(i)));
				
				// 연산자 만났을 때
				if(stack.peek().equals("*")) {
					temp = stack.pop();
					continue;
				}
				else if(stack.peek().equals("+")) {
					continue;
				}
				// 숫자인 경우
				else {
					if(temp.equals("*")) {
						int sum = Integer.parseInt(stack.pop())*Integer.parseInt(stack.pop());
						stack.add(String.valueOf(sum));
						temp = "";
					}
					else {
						continue;
					}
				}
			}
			
			int sum =0;
			int size= stack.size();
			for(int i=0; i<size; i++) {
				// 연산자 만났을 때
				if(stack.peek().equals("+")) {
					stack.pop();
					continue;
				}
				// 숫자인 경우
				else {
					 sum += Integer.parseInt(stack.pop());
				}
			}
			
			sb.append(sum);
			sb.append("\n");
		}
		System.out.println(sb);
	}
}

🤔 나의 생각

이 문제도 구조는 동일하다. 그러나 “*“이 추가되었다.
전체 문자열을 순서대로 STACK에 받아와 연산자를 만나면 pop 하여 바로 계산해주도록 하였다.
여기서 주의할 점은 연산순서이다. 곱셈이 덧셈보다 먼저이기 때문에
나는 곱셈 연산을 다해준 뒤 남은 것을 가지고 덧셈연산을 해주었다.
덧셈 연산자 부분에서 stack.size()가 pop때문에 계속 변한다는 것을 생각하지 못해서 애를 먹었다.
그래서 size라는 변수를 통해 처음 사이즈로 고정시켜 주었다.