[백준-17413] 단어 뒤집기 2 / Python

17413번: 단어 뒤집기 2

import sys

input = sys.stdin.readline

s = input().rstrip()
flag = False
word = ""
answer = ""

for i in s:
    if flag == False:
        if i == "<":
            flag = True
            word += i
        elif i == " ":
            word += i
            answer += word
            word = ""
        else:
            word = i + word

    else:
        word += i
        if i == ">":
            flag = False
            answer += word
            word = ""

print(answer + word)

 

해설

flag를 이용해 풀 수 있었다. '<'를 만나는 순간 flag를 True로 전환하여 문자열 그대로를 answer에 복사한다. flag가 False인 상태이면서 '<   >' 사이의 단어가 아니라면 뒤집어 answer에 저장한다.