백준 17413번 : 단어 뒤집기 2(Java)

https://www.acmicpc.net/problem/17413

17413번: 단어 뒤집기 2

---

[풀이]

• 스택 자료구조를 이용해서 풀었습니다.
• <>안의 문자열은 그대로 출력, 이외의 문자열은 공백을 기준으로 역순으로 출력해야합니다. 이 때 꺽쇠 문자열이라면 flag 변수를 이용해서 표시합니다.
• '>'혹은 <> 안의 문자열은 별다른 조치를 하지 않고 출력을 하며 공백이 나타날 경우 스택에 쌓여있던 문자열을 역순으로 출력합니다.
• 마지막 공백 이후의 문자열은 스택에 담겨지기만 할 뿐 출력이 되지 않기 때문에 따로 출력을 해야합니다.

---

[코드]

 

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String input = br.readLine();
        Stack<Character> stack = new Stack<>();
        boolean flag = false; // <> 안에 있는지 표시

        for (int i = 0; i < input.length(); i++) {
            if (input.charAt(i) == '<') {
                flag = true;
                while (!stack.isEmpty()) {
                    System.out.print(stack.pop());
                }
                System.out.print(input.charAt(i));
            } else if (input.charAt(i) == '>') {
                flag = false;
                System.out.print(input.charAt(i));
            } else if (flag) {
                System.out.print(input.charAt(i));
            } else {
                if (input.charAt(i) == ' ') {
                    while (!stack.isEmpty()) {
                        System.out.print(stack.pop());
                    }
                    System.out.print(" ");
                } else
                    stack.push(input.charAt(i));
            }
        }
        while (!stack.isEmpty()) {
            System.out.print(stack.pop());
        }
    }
}