[백준 | 파이썬] 17413 : 단어 뒤집기 2

💡 접근 방식은 쉽게 떠올릴 수 있었으나, 구현이 조금 까다로웠던 문제

문제

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

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

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

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

입력

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

출력

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

풀이 방법

while 문 안에서 idx 를 증가시켜가면서 조건과 일치하는지 확인한다.

1. '<'를 만날 경우, 인덱스를 하나 증가시키고 '>'를 만날 때까지 인덱스를 증가시킨다. '>'를 만나면 인덱스를 한번 더 증가시킨다.
2. 알파벳이나 숫자를 만날 경우, 현재 인덱스를 시작 인덱스로 설정한다. 인덱스가 단어 길이를 넘지 않으면서 공백을 만나기 전까지 인덱스를 증가시킨다. 공백을 만나면 문장을 시작 인덱스부터 현재 인덱스까지로 범위를 설정해 슬라이싱한다. 슬라이싱한 문자열은 반전시켜서 기존 문장의 인덱스 자리에 집어넣는다.
3. 공백을 만나면 인덱스를 한번 증가시킨다.

# '단어' 만 뒤집는다
S = list(input())

i = 0
start = 0

while i < len(S):
    if S[i] == '<':
        i += 1
        while S[i] != '>':
            i += 1
        i += 1
    elif S[i].isalnum():
        start = i
        while i < len(S) and S[i].isalnum():
            i += 1
        temp = S[start:i]
        temp.reverse()
        S[start:i] = temp
    else:
        i += 1

print(*S, sep="")