BOJ: 단어 뒤집기2 - 17413번

17413번: 단어 뒤집기 2

 

문제

문자열 S가 주어졌을 때, 이 문자열에서 단어만 뒤집으려고 한다.
먼저, 문자열 S는 아래와과 같은 규칙을 지킨다.
알파벳 소문자('a'-'z'), 숫자('0'-'9'), 공백(' '), 특수 문자('<', '>')로만 이루어져 있다.
문자열의 시작과 끝은 공백이 아니다.
'<'와 '>'가 문자열에 있는 경우 번갈아가면서 등장하며, '<'이 먼저 등장한다. 또, 두 문자의 개수는 같다.
태그는 '<'로 시작해서 '>'로 끝나는 길이가 3 이상인 부분 문자열이고, '<'와 '>' 사이에는 알파벳 소문자와 공백만 있다. 단어는 알파벳 소문자와 숫자로 이루어진 부분 문자열이고, 연속하는 두 단어는 공백 하나로 구분한다. 태그는 단어가 아니며, 태그와 단어 사이에는 공백이 없다.

 

내가 작성한 코드

# https://www.acmicpc.net/problem/17413
# 단어 뒤집기2
import sys
from collections import deque

word = sys.stdin.readline().strip()
stack = deque()

for s in word:
    if s == "<":
        while len(stack) > 0:
            print(stack.pop(), end="")
        stack.append(s)
    elif s == ">":
        stack.append(s)
        while len(stack) > 0:
            print(stack.popleft(), end="")
    elif s == " " and stack[0] != "<":
        while len(stack) > 0:
            print(stack.pop(), end="")
        print(" ", end="")
    else:
        stack.append(s)

while len(stack) > 0:
    print(stack.pop(), end="")
print()

 

• collections의 deque를 이용해서 풀었다. 그런데 정작 변수명은 stack
• <는 태그의 시작점을 뜻한다. <를 만나면 태그 안의 내용이 끝났다는 것을 뜻하기 때문에, stack에 요소가 들어있다면 모두 출력한다. 그 다음에 <를 stack에 넣어준다.
• >는 태그의 끝을 뜻한다. stack에 들어있는 모든 문자를 출력하되, deque의 popleft()를 활용해서 앞에서부터 출력한다. 왜냐하면 문제에서 태그는 단어가 아니기 때문에 뒤집지 않아야한다고 명시했기 때문이다.
• 해당 문자가 공백이라면, stack에 첫번째 요소가 <인지 확인한다. 태그 안의 공백은 무시하고 넘어가야하기 때문이다. 단어라면 pop()을 이용해서 뒤집어서 출력한다.
• 그 외의 경우는 태그 내용이거나, 단어이기 때문에 stack에 추가한다.
• word탐색이 끝났다면, stack이 빌 때까지 뒤집어서 출력한다.