[백준] 17413. 단어 뒤집기2

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

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

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

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

---

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws Exception {
        Queue<StringBuilder> queue = new LinkedList<>();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), "\n");

        StringBuilder sb = new StringBuilder(st.nextToken());
        StringBuilder sb3 = new StringBuilder();
        int k = -1;
        int j = -1;
        for(int i = 0; i < sb.length(); i++) {
            if(sb.charAt(i) == '<') {
                j = i;
            } else if (sb.charAt(i) == '>') {
                k = i;
            }
            if (j != -1 && k != -1) {
                queue.add(sb3);
                sb3 = new StringBuilder();
                StringBuilder sb2 = new StringBuilder();
                sb2.append(sb.substring(j, k + 1));
                sb2.reverse();
                queue.add(sb2);
                j = -1;
                k = -1;
            } else if (j == -1 && k == -1) {
                sb3.append(sb.charAt(i));
            } else if (sb.charAt(i) == ' ') {
                queue.add(sb3);
                sb3 = new StringBuilder();
            }
        }

        queue.add(sb3);
        sb3 = new StringBuilder();
        while(!queue.isEmpty()) {
            if(queue.peek().equals("")) {
                queue.poll();
                continue;
            }
            if(queue.peek().indexOf("<") < 0) {
                String[] str = queue.poll().toString().split(" ");
                for(int i = 0; i < str.length; i++) {
                    StringBuilder tsb = new StringBuilder(str[i]);
                    sb3.append(tsb.reverse()).append(" ");
                }
                sb3.deleteCharAt(sb3.length()-1);
            } else {
                sb3.append(queue.poll().reverse());
            }
        }
        System.out.println(sb3.toString());
    }
}

시작 인덱스(k)와 끝 인덱스(j)를 지정하고 괄호가 시작되는 부분과 끝나는부분의 인덱스를 저장한다.

 

시작 인덱스와 끝 인덱스가 정해지지 않았으면(j == -1 && k == -1) 문자열이나 공백이 있다는것이니 StringBuilder sb3에 붙인다. 공백이 나타나면(sb.charAt(i) == ' ') 지금까지 저장된 sb3을 큐에 넣고 새로 초기화를 해준다.

 

괄호의 시작 인덱스와 끝 인덱스가 정해지면(j != -1 && k != -1)  큐에 괄호 사이의 문자 값을 저장한 sb3을 넣고 다시 초기화를 한다. 새로운 StringBuilder sb2를 만들어 입력받은 문자열의 시작 인덱스부터 끝 인덱스까지 자르고 뒤집어 큐에 넣는다. 그리고 다음 괄호를 찾기위해 인덱스를 -1로 초기화한다.

 

큐가 비어있지 않을때까지 반복문을 돌면서 큐에 들어간 값에 "<"문자가 없으면(문자열로만 구성되면) 해당 값을 공백으로 split 한 다음 뒤집어 StringBuilder에 붙이고 공백도 append해준다.