백준: 17413 단어 뒤집기 2

문제

17413번: 단어 뒤집기 2

 

문제 풀이

 9039 단어 뒤집기는 단순히 뒤집기만 하면 되므로 심플 그자체이다. 하지만 단어 뒤집기 2는 < >의 사이에 있는 태그 문자의 경우 반전을 하지 않으며 나머지는 반전을 하여야 한다. 문제는 다음과 같이 접근할 수 있다.

 

• reverse 플래그를 통해 반전 여부를 판단한다.
• '<' 가 입력되면, 반전이 불필요 하므로 reverse = False로 설정한다.
• '>' 가 입력되면 반전이 필요할 수도 있으므로 reverse = True로 설정한다.
• reverse의 초기값은 True로 설정하여 '<' 가 입력되지 않는다면 바로 반전될 수 있도록 한다.

 stack으로 분류된 문제인데 이 문제는 해결하기 위해 stack을 사용하지 않았다.  '<'가 입력되면 stack에 추가하여 문제를 해결할 수 도 있지만, 올바른 괄호 찾기 처럼 여러번 중첩되는 상황이 없기 때문에 굳이 stack을 사용하지 않아도 되며, stack에 push, pop을 하는 것은 불필요한 오버헤드이다.😐

 

코드

from sys import stdin


if __name__ == "__main__":
    s = stdin.readline().rstrip()
    reverse = True
    answer = ''
    word = ''
    for c in s:
        if c == '<':
            reverse = False
            answer += word
            word = '<'
        elif c == '>':
            reverse = True
            answer += (word + '>')
            word = ''
        elif c == ' ':
            answer += (word + ' ')
            word = ''
        elif reverse:
            word = c + word
        else:
            word += c

    print(answer + word)