[Java] 백준 17413 - 단어 뒤집기 2

Baekjoon_17413 - 단어 뒤집기 2

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

문제

문자열 S가 주어졌을 때, 이 문자열에서 단어만 뒤집으려고 한다.

먼저, 문자열 S는 아래와과 같은 규칙을 지킨다.

1. 알파벳 소문자('a'-'z'), 숫자('0'-'9'), 공백(' '), 특수 문자('<', '>')로만 이루어져 있다.
2. 문자열의 시작과 끝은 공백이 아니다.
3. '<'와 '>'가 문자열에 있는 경우 번갈아가면서 등장하며, '<'이 먼저 등장한다. 또, 두 문자의 개수는 같다.

태그는 '<'로 시작해서 '>'로 끝나는 길이가 3 이상인 부분 문자열이고, '<'와 '>' 사이에는 알파벳 소문자와 공백만 있다. 단어는 알파벳 소문자와 숫자로 이루어진 부분 문자열이고, 연속하는 두 단어는 공백 하나로 구분한다. 태그는 단어가 아니며, 태그와 단어 사이에는 공백이 없다.

입력

첫째 줄에 문자열 S가 주어진다. S의 길이는 100,000 이하이다.

출력

첫째 줄에 문자열 S의 단어를 뒤집어서 출력한다.

---

해결방법

• 주어진 단어를 한 글자씩 돌며 단어면 스택에 넣고 pop() 하면서 뒤집어서 StringBuilder에 추가하고, 태그면 그냥 추가한다.
• 단어를 순회하다 만난 글자가 < 이면 태그의 시작을 의미하므로 isTag 를 true로, > 이면 태그의 끝을 의미하므로 isTag를 false로 바꿔가면서 태그인지 단어인지 구분한다.

코드

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

public class Baekjoon_17413 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String inputString = br.readLine();
        Stack<Character> stack = new Stack<>();
        StringBuilder sb = new StringBuilder();
        boolean isTag = false;

        for (int i = 0; i < inputString.length(); i++) {
           if(inputString.charAt(i) == '<'){
               while(!stack.isEmpty()){
                   sb.append(stack.pop());
               }
               isTag = true;
           } else if (inputString.charAt(i) == '>') {
               isTag = false;
               sb.append(inputString.charAt(i));
               continue;
           }

            if (isTag) {
                sb.append(inputString.charAt(i));
            }else{
                if (inputString.charAt(i) == ' ') {
                    while (!stack.isEmpty()) {
                        sb.append(stack.pop());
                    }
                    sb.append(' ');
                    continue;
                }else{
                    stack.push(inputString.charAt(i));
                }
            }

            if(i == inputString.length() - 1){
                while (!stack.isEmpty()) {
                    sb.append(stack.pop());
                }
            }
        }

        System.out.println(sb);
    }

}