[Baekjoon] 17413번 단어 뒤집기 2

문제

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

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

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

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

입력

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

출력

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

예제

Example 1:

Input: 
baekjoon online judge
Output: 
noojkeab enilno egduj

Example 2:

Input:
<open>tag<close>
Output:
<open>gat<close>

Example 3:

Input:
<ab cd>ef gh<ij kl>
Output:
<ab cd>fe hg<ij kl>

Example 4:

Input:
one1 two2 three3 4fourr 5five 6six
Output:
1eno 2owt 3eerht rruof4 evif5 xis6

Example 5:

Input:
<int><max>2147483647<long long><max>9223372036854775807
Output:
<int><max>7463847412<long long><max>7085774586302733229

Example 6:

Input:
<problem>17413<is hardest>problem ever<end>
Output:
<problem>31471<is hardest>melborp reve<end>

Example 7

Input:
<   space   >space space space<    spa   c e>
Output:
<   space   >ecaps ecaps ecaps<    spa   c e>

조건

시간 제한 : 1초

메모리 제한 : 512 MB

풀이과정

내 풀이

s = input()

st = []
tag = False
for i in s:
  if tag == False:
    if i ==" ":
      while(st):
        print(st.pop(),end='')
      print(" ", end='')
    elif i == "<":
      while(st):
        print(st.pop(),end='')
      tag = True
      print("<",end='')
    else:
      st.append(i)
  else:
    print(i,end= '')
    if i == ">":
      tag = False

while(st):
  print(st.pop(),end='')

풀이의 핵심은 다음과 같다.

1. tag가 진행중인지 여부를 위한 변수 tag를 두고 이를 기준으로 나누어서 진행한다.
2. tag가 진행중이지 않고 이 때 문자가 공백 혹은 “<”가 아니라면 스택에 쌓는다.
3. 공백혹은 “<”를 만나면 스택을 모두 출력한다.

스택은 선입후출 자료구조로 뒤집어야 하는 경우 사용하기에 좋다고 판단하였다.