백준 1918 후위표기식 자바[java]

• 후위 표기식

1918번: 후위 표기식

 

연산기호만 스택에 넣는 접근은 좋았다.

근데 연산기호의 우선순위에 따라 출력할지 말지 정하는 부분은 미흡했다.

그래서 다른 블로그를 참고했다.

 

어찌저찌해서 정답의 흐름대로 잘 짰는데,

가운데 부분에 break;를 쓰지 않아서 계속 틀렸었다,,,

근데 왜 break 써야하는지 아직 잘 모르겠다.

문제 간단설명)

중위표기식을 후위표기식으로 바꾸기

 

풀이)

문자는 그대로 출력하고,

'(' 가 나오면 스택에 넣는다.

')' 가 나오면 스택에서 '(' 가 나올때까지 pop()해주고

나머지 연산기호들이 나오면 우선순위에 따라 출력하거나 스택에 넣는다.

 

import java.io.*;
import java.util.*;
public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();
        Stack<Character> stk = new Stack<>();
        String str = br.readLine();
        for (int i = 0; i < str.length(); i++) {
            char ch = str.charAt(i);
 
            if ('A' <= ch && ch <= 'Z') {
                sb.append(ch);
            } else if (ch == '(') {
                stk.add(ch);
            } else if (ch == ')') {
                while (!stk.isEmpty()) {
                    if (stk.peek() == '(') {
                        stk.pop();
                        break;
                    }
                    sb.append(stk.pop());
                }
            } else {
                while (!stk.isEmpty() && getPriority(stk.peek()) >= getPriority(ch)) {
                    sb.append(stk.pop());
                }
                stk.add(ch);
            }
        }
        while (!stk.isEmpty())
            sb.append(stk.pop());
        bw.write(sb.toString());
        bw.flush();
        bw.close();
    }
 
    public static int getPriority(char c) {
        if (c == '(')
            return 0;
        else if (c == '+' || c == '-')
            return 1;
        else
            return 2;
    }
}