[문제] 백준 1935번 후위 표기식2

* 출처: 백준

www.acmicpc.net/problem/1935

1935번: 후위 표기식2

 

 

내 풀이

 

calc()인자로 alpha 배열을 안넘겨주고 풀어서 틀렸었다.

public class Main {

	static int N;

	public static void main(String[] args) throws IOException {
		BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(in.readLine()); // 피연산자 개수
		String s = in.readLine(); // 후위 연산식

		int[] alpha = new int[N]; // 피연산자 대응값
		for (int i = 0; i < N; i++) {
			alpha[i] = Integer.parseInt(in.readLine());
		}

//		calc(s,alpha);
		System.out.printf("%.2f", calc(s, alpha));

	}

	static double calc(String s, int[] alpha) {
		int len = s.length();
		double res = 0;
		Stack<Double> stack = new Stack<>();
		for (int i = 0; i < len; i++) {
			char c = s.charAt(i);

			if (c >= 'A' && c <= 'Z') {
				int cIdx = c - 'A';
				double tmp = alpha[cIdx];
				stack.push(tmp);

			} else if (c == '+') {
				double b = stack.pop();
				double a = stack.pop();
				res = a + b;
				stack.push(res);
			} else if (c == '-') {
				double b = stack.pop();
				double a = stack.pop();
				res = a - b;
				stack.push(res);
			} else if (c == '*') {
				double b = stack.pop();
				double a = stack.pop();
				res = a * b;
				stack.push(res);
			} else if (c == '/') {
				double b = stack.pop();
				double a = stack.pop();
				res = a / b;
				stack.push(res);
			}

		}
		res = stack.pop();
		return res;

	}
}