[Python] 백준 17413번 : 단어 뒤집기 2

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

17413번: 단어 뒤집기 2

---

문제 풀이

문자열 문제이다.

 

flag를 사용해서 True 상태라면 그대로 입력받고 아니라면 스택에 추가한다.

<이 등장하면 flag 상태를 True로 바꿔주고 >가 등장하면 False로 바꿔준다.

스택에 있는 단어를 넣어주는 경우는 2가지로, <이 등장할 때와 ' ' (공백)이 등장할 때이다.

느낀 점

IDE이 없이 백준 페이지에서 풀어서 코드가 깔끔 치 못하다. 조금 아쉽다.

코드

import sys

S = sys.stdin.readline().strip()
stack = []
result = ''
flag = False

for ch in S:
  if ch == '<':
    if stack:
      while stack:
        result += stack.pop()
    result += ch
    flag = True
  elif ch == '>':
    result += ch
    flag = False
  elif ch == ' ':
    if stack:
      while stack:
        result += stack.pop()
    result += ch
  if ch not in ['<', '>', ' '] and flag: result += ch
  elif ch not in ['<', '>', ' '] and  not flag: stack.append(ch)

if stack:
  while stack:
    result += stack.pop()
      
print(result)