[백준] 17413 단어 뒤집기 2

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

17413번: 단어 뒤집기 2

 

입력받은 문자열을 하나씩 탐색해 가면서 <> 태그 안에 있다면 문자 그대로 출력하고, 태그 밖이라면 스택에 쌓은 후 꺼내서 출력해 주어 해결할 수 있다.

 

태그 안밖을 구분하기 위해 boolean형 변수를 사용할 수 있다.

import java.util.*;
import java.io.*;

// 단어 뒤집기 2 https://www.acmicpc.net/problem/17413
class No17413 {
    public static void main(String args[]) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String input = br.readLine();

        Stack<Character> s = new Stack<>();

        // 태그 안밖 구분 (false면 밖, true면 안)
        boolean flag = false;

        for (int i = 0; i < input.length(); i++) {

            if (input.charAt(i) == '<') {   // 태그 시작
                flag = true;
                while (!s.isEmpty()) {
                    System.out.print(s.pop());
                }
                System.out.print("<");
            } else if (input.charAt(i) == '>') {    // 태그 끝
                flag = false;
                System.out.print(">");
            } else if (!flag) { // 태그 밖
                if (input.charAt(i) == ' ') {   // 공백을 만나면
                    while (!s.isEmpty()) {
                        System.out.print(s.pop());  // 스택 다 출력
                    }
                    System.out.print(' ');
                }
                if (input.charAt(i) != ' ') {
                    s.add(input.charAt(i));
                }
            } else if (flag) {  // 태그 안
                System.out.print(input.charAt(i));
            }
        }
        while (!s.isEmpty()) {
            System.out.print(s.pop());
        }

    }
}