[백준/17413] 단어 뒤집기 2 (파이썬/python)

1️⃣ 문제

문제 링크 : https://www.acmicpc.net/problem/17413

2️⃣ 코드

from collections import deque

flag=0
ans=''
word=''

ary=input()


for i in ary:
    if i=='<':
        flag=1
        ans+=word[::-1]
        word=''
        ans+=i

    elif i=='>':
        flag=0
        ans+=i
        
    elif flag==1:
        ans+=i

    elif i==' ':
        ans+=word[::-1]
        ans+=' '
        word=''

    elif flag==0:
        word+=i


ans+=word[::-1]
print(ans)