[Python] 백준 / 단어 뒤집기 2 / 17413번 / 문자열

문제

단어 뒤집기 2 문제 링크

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

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

알파벳 소문자('a'-'z'), 숫자('0'-'9'), 공백(' '), 특수 문자('<', '>')로만 이루어져 있다.
문자열의 시작과 끝은 공백이 아니다.
'<'와 '>'가 문자열에 있는 경우 번갈아가면서 등장하며, '<'이 먼저 등장한다. 또, 두 문자의 개수는 같다.
태그는 '<'로 시작해서 '>'로 끝나는 길이가 3 이상인 부분 문자열이고, '<'와 '>' 사이에는 알파벳 소문자와 공백만 있다. 단어는 알파벳 소문자와 숫자로 이루어진 부분 문자열이고, 연속하는 두 단어는 공백 하나로 구분한다. 태그는 단어가 아니며, 태그와 단어 사이에는 공백이 없다.

입력

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

baekjoon online judge

<ab cd>ef gh<ij kl>

출력

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

noojkeab enilno egduj

<ab cd>fe hg<ij kl>

접근 방식

1. '>'를 '<'로 바꾸고 split('<')을 하게 되면 태그 안의 문자들은 무조건 리스트의 홀수인덱스에 존재하게 된다 ( 1, 3, 5, 7번째에 존재 )

2. 리스트의 홀수번째는 공백을 기준으로 split 후 각 단어들을 뒤집어서 join 해준다

코드

strs = input()
answer = ''
li = strs.replace('>','<').split('<')

for i in range(len(li)):
    if i % 2 != 0 :
        answer += '<'+ li[i] + '>'
    else :
        non_tag_li = li[i].split(' ')
        answer += ' '.join([word[::-1] for word in non_tag_li])
    
print(answer)