[백준/파이썬] 17413번 단어 뒤집기 2

-Code

s = input()
result = ""
stack = ""
flag = False
for word in s:
    if not flag:
        if word == '<':
            flag = True
            result += stack[::-1]
            stack = ""
            stack += word
        elif word == ' ':
            result += stack[::-1] + word
            stack = ""
        else:
            stack += word
    else:
        stack += word
        if word == '>':
            flag = False
            result += stack
            stack = ""

print(result + stack[::-1])

처음에 flag를 이용해 작성했으나 if문이 섞여서 <>가 들어간 부분에서 문제가 생겨서 위치를 조금 씩 옮기면서 고치다 도저히 안되어서 찾아보니 flag if문 안에 다른 if문을 넣고 조금 고치니 해결이 되었습니다.