[BOJ] 17413 - 단어 뒤집기 2

개인적인 풀이일 뿐, 최적의 정답이 아님을 알려드립니다.

 

문제

www.acmicpc.net/problem/17413

17413번: 단어 뒤집기 2

난이도: 실버 3

사용언어: JAVA

 

풀이

이 문제는 '<', '>', 그리고 공백에 대한 예외처리만 잘해주면 쉽게 해결할 수 있습니다. 저는 스택을 이용해 문자열을 뒤집고 출력했습니다.

 

import java.io.*;
import java.util.Stack;
 
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));
 
        Stack<Character> st = new Stack<>();
        String s = br.readLine();
        int i;
        for (i = 0; i < s.length(); i++) {
            if (s.charAt(i) == '<') { // '<'인 경우, 앞의 문자열을 뒤집어 출력해주고,
                while (st.size() != 0) {
                    bw.write(st.pop());
                }
                while (s.charAt(i) != '>') { // '<'이 나올때 까지 그냥 출력
                    bw.write(s.charAt(i));
                    i++;
                }
                bw.write(s.charAt(i));
            } else if (s.charAt(i) == ' ') { // ' '인 경우, 앞까지의 문자열을 뒤집어 출력해주고
                while (st.size() != 0) {
                   bw.write(st.pop()); // 모두 출력했으면 공백 출력
                }
                bw.write(s.charAt(i));
            } else {
                st.push(s.charAt(i));
            }
        }
 
        while (st.size() != 0) { // 마지막 문자열 출력
            bw.write(st.pop());
        }
 
        bw.flush();
    }
}