(백준) 17413 파이썬 - 단어 뒤집기 2

백준 17413 - 단어 뒤집기 2

문제

문자열 S가 주어졌을 때, 이 문자열에서 단어만 뒤집으려고 한다.

먼저, 문자열 S는 아래와과 같은 규칙을 지킨다.

1. 알파벳 소문자('a'-'z'), 숫자('0'-'9'), 공백(' '), 특수 문자('<', '>')로만 이루어져 있다.
2. 문자열의 시작과 끝은 공백이 아니다.
3. '<'와 '>'가 문자열에 있는 경우 번갈아가면서 등장하며, '<'이 먼저 등장한다. 또, 두 문자의 개수는 같다.

태그는 '<'로 시작해서 '>'로 끝나는 길이가 3 이상인 부분 문자열이고, '<'와 '>' 사이에는 알파벳 소문자와 공백만 있다. 단어는 알파벳 소문자와 숫자로 이루어진 부분 문자열이고, 연속하는 두 단어는 공백 하나로 구분한다. 태그는 단어가 아니며, 태그와 단어 사이에는 공백이 없다.

입력

첫째 줄에 문자열 S가 주어진다. S의 길이는 100,000 이하이다.

출력

첫째 줄에 문자열 S의 단어를 뒤집어서 출력한다.

코드

import re

answer = []
S = input() # 문자열을 입력받는다.

def reverse_string(input_string):
    ret = ""
    for i in input_string.split():
        ret += "".join(reversed(i))
        ret += " "
    return ret.rstrip()

pattern = re.compile("(<.+?>)|([^<>]+)") 
arr = re.split(pattern, S)       # 패턴에 맞게 split한 배열을 사용

for word in arr:
    if word == "" or word is None:
        continue
    if "<" in word:
        # "<"를 포함한 word는 그냥 정답배열에 넣는다
        answer.append(word) 
    else:
        # 이외의 word는 뒤집어서 넣는다.
        answer.append(reverse_string(word)) 

for i in answer:
    print(i, end="")

RegExr 에서 정규표현식에 대해서 많은 시도 끝에 적절한 패턴을 찾을 수 있었습니다. 그 이외에는 그렇게 어려운 부분이 없었다고 생각합니다.