[Python] 백준 17413번

17413. 단어 뒤집기2 (silver3)

# 자료구조

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

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

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

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

 

입력

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

 

출력

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

 

 

예제 입력 1

baekjoon online judge

예제 출력 1

noojkeab enilno egduj

 

예제 입력 2

<open>tag<close>

예제 출력 2

<open>gat<close>

 

예제 입력 3

<ab cd>ef gh<ij kl>

예제 출력 3

<ab cd>fe hg<ij kl>

 

 

 

 

풀이) 

 

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

주어진 규칙에 따라 while문 안에 코드를 작성해주었다.

먼저 words라는 리스트를 만들고 입력을 받았다. 그 다음 while문을 사용해서 words 리스트의 길이만큼 반복해주었다.

if문을 사용해서 words 리스트의 원소 중 열린 괄호를 만났을 때는 닫힌 괄호를 만나기 전까지 뒤집지 않고 지나가야 하기 때문에 인덱스에 해당하는 idx를 증가시켜주었다. 닫힌 괄호를 만났을 때도 넘어가야하기 때문에 인덱스(idx)를 1 증가시키고, 다음 원소에 대해 검사하기 위해 인덱스를 또 1 증가시켜주었다.

elif문과 isalnum()함수를 사용해서 words 리스트의 원소가 숫자 또는 알파벳일 때 검사를 해주었다.

검사하는 첫 번째 원소의 인덱스에 해당하는 start를 숫자/알파벳을 발견한 인덱스인 idx로 저장해주었고, 

while문을 또 사용해서 idx가 words 리스트 범위 내이고, 리스트 원소가 숫자, 알파벳일 때 idx의 값을 1씩 증가시켜주었다.

temp라는 변수를 사용해서 start부터 idx, 즉 숫자/알파벳인 범위의 부분만 reverse()함수를 사용해서 뒤집어준 걸 저장해주었다.

그 다음 temp 변수에 저장된 값을 다시 words리스트에 저장해주었다.

그 외의 경우에 해당하는 공백의 경우는, else문을 사용해서 idx를 1 증가시켜서 지나가도록 처리해주었다.

마지막으로 while문 밖에서 words 리스트를  join을 사용해서 원소 구분 없이 출력해주면 된다.

import sys
input = sys.stdin.readline

words = list(input().rstrip())

idx = 0
start = 0
while idx < len(words):
    if words[idx] == "<":           # 열린 괄호를 만났을 때
        idx += 1 
        while words[idx] != ">":     # 닫힌 괄호를 만날 때 까지
            idx += 1 
        idx += 1               
    elif words[idx].isalnum():  # 숫자/알파벳이 있을 때
        start = idx             
        while (idx < len(words)) and (words[idx].isalnum()):
            idx += 1
        temp = words[start:idx]  # 해당 인덱스까지 뒤집기
        temp.reverse()       
        words[start:idx] = temp  # 뒤집은 걸 다시 저장해줌
    else:                   # 공백
        idx+=1                

print("".join(words))