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

문제 링크

https://www.acmicpc.net/problem/17413

17413번: 단어 뒤집기 2

 

문제 풀이

1. '<알파벳 소문자/공백>' 또는 '알파벳 소문자/숫자/공백'으로 이루어진 문자열을 찾아 words 변수에 담는다. (re.compile을 사용하여 정규 표현식을 컴파일)
2. words를 순회하며 조건에 따라 result 변수에 이어 붙인다.
   1. 만약 '<'가 포함되어 있다면 그대로 이어 붙인다.
   2. '<'가 포함되어 있지 않고, ' '(공백)이 있다면 공백을 기준으로 단어를 잘라서 역순하여 이어 붙인다.
   3. '<'가 포함되어 있지 않고, ' '(공백)이 없다면 단어를 바로 역순하여 이어 붙인다.

 

전체 코드

import re
from sys import stdin
input = stdin.readline

str = input().strip()
p = re.compile('<[a-zA-Z0-9 ]+>|[a-zA-Z0-9 ]+')
words = p.findall(str)
result = ''

for word in words :
    if word[0] == '<' :
        result += word
    else :
        if ' ' in word :
            s = word.split()
            for i in range(len(s)) :
                result += ''.join(reversed(s[i]))
                if i != len(s)-1 : result += ' '
        else :
            result += ''.join(reversed(word))

print(result)