[17413] 단어 뒤집기 2

🔗 문제 링크

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

❔ 문제 설명

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

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

• 알파벳 소문자('a'-'z'), 숫자('0'-'9'), 공백(' '), 특수 문자('<', '>')로만 이루어져 있다.

• 문자열의 시작과 끝은 공백이 아니다.

• '<'와 '>'가 문자열에 있는 경우 번갈아가면서 등장하며, '<'이 먼저 등장한다. 또, 두 문자의 개수는 같다.

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

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

⚠️ 제한사항

• 첫째 줄에 문자열 S가 주어진다.

• S의 길이는 100,000 이하이다.

---

💡 풀이 (언어 : Java)

뒤집어서 출력하는 것은 들어오는 것의 반대 순서로 출력하는 LIFO다. 즉 스택을 이용해서 구현하면 된다.

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        char[] input = br.readLine().toCharArray();
        StringBuilder answer = new StringBuilder();
        Stack<Character> stack = new Stack<>();
        // 태그 안인지 알려주는 변수
        boolean flag = false;

        for (char ch : input) {
            // 태그 시작
            if (ch == '<') {
                // 앞에 글자들 아직 뒤집어서 정답에 반영 안했으면 처리
                while (!stack.isEmpty())
                    answer.append(stack.pop());
                answer.append('<');
                flag = true;
            // 태그 종료
            } else if (ch == '>') {
                answer.append('>');
                flag = false;
            // 태그 안의 내용은 그대로 넣기
            } else if (flag) {
                answer.append(ch);
            // 태그 밖일 때, 공백 만났을 경우
            } else if (ch == ' ') {
                // 공백 직전까지 글자를 뒤집어서 넣어줌 = 스택에서 pop()
                while (!stack.isEmpty())
                    answer.append(stack.pop());
                answer.append(' ');
            // 태그 밖에서 공백아닌 글자들은 전부 스택에 넣어줌
            } else {
                stack.push(ch);
            }
        }
        // for문을 다돌았을때 아직 처리안된 stack의 글자들을 뒤집어서 넣기
        while (!stack.isEmpty())
            answer.append(stack.pop());
        System.out.print(answer);
    }
}