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

문제

17413번: 단어 뒤집기 2

입력

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

출력

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

입력 예제

baekjoon online judge

<ab cd>ef gh<ij kl>

출력 예제

noojkeab enilno egduj

<ab cd>fe hg<ij kl>

문제 풀이

• 괄호를 만나는 경우와 문자나 숫자를 만나는 경우, 공백을 만나는 경우 세 가지로 나눠 탐색한다.
  
  • 열린괄호(<)를 만나는 경우
    • 닫힌 괄호(>)를 만날 때 까지 인덱스 증가시킨다.
    • 닫힌 괄호를 만났다면, 마지막으로 인덱스를 증가시킨다.(문자나 숫자인 경우 탐색하기 위해서)
  • 문자나 숫자를 만나는 경우
    • 문자나 숫자의 시작 인덱스에 증가시켜온 인덱스 값을 넣어 시작 인덱스를 구한다.
    • 그 후, 문자나 숫자인 경우 계속해서 인덱스를 증가시켜 끝 인덱스를 구한다.
    • 임시 변수를 생성한 후, 시작 인덱스부터 끝 인덱스 까지 문자를 뒤집은 후, 원래 문자의 값에 넣는다.
  • 공백을 만나는 경우
    • 인덱스를 증가시킨다.

코드

word = list(input())
idx = 0
start = 0

# 인덱스 번호로 탐색
while idx < len(word):
    # 열린 괄호 만나는 경우
    if word[idx] == '<':
        idx += 1
        # 닫힌 괄호 만날 때 까지 안에 있는 글자 그대로
        while word[idx] != '>':
            idx += 1
        idx += 1

    # 숫자나 문자 만나는 경우 시작 인덱스, 끝 인덱스 구해서 뒤집기
    elif word[idx].isalnum():
        start = idx
        while idx < len(word) and word[idx].isalnum():
            idx += 1

        tmp = word[start:idx]
        tmp.reverse()
        word[start:idx] = tmp

    # 공백인 경우 인덱스 증가
    else:
        idx += 1
print(''.join(word))