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

문제

 

내 코드

s = input()

word = []
w = ''
idx = 0
while(idx < len(s)):
    if s[idx] == '<':
        if w:
            word.append(''.join(reversed(w)))
            w = ''
        for i in range(idx+1 , len(s)):
            if s[i] == '>':
                word.append(s[idx:i+1])
                idx = i
                break
    elif s[idx] == ' ':
        word.append(''.join(reversed(w)))
        w = ''
        word.append(' ')
    else:
        w = w+s[idx]
    idx += 1

if w:
    word.append(''.join(reversed(w)))
print(''.join(word))

 

 처음엔 일단 괄호가 나오면 스택인가 싶어 스택을 이용해 구현할 수 있는지 고민해보았다.

그런데 괄호로만 이루어진 단어가 아니기 때문에 다른 방법을 생각해보았다.

 

<> 괄호가 나오는 태그인 경우와 띄어쓰기가 있는 단어를 나누어 생각해보았다.

 

일단 '<>' 태그인 경우를 먼저 처리한다.

1. '<'를 만났을 때
2. w에 저장된 단어가 있다면 w를 역순으로 바꾸어 결과값일 리스트 word에 넣어준다.
3. '<' 뒤부터 문자열 S를 for문을 이용해 제일 가까운 '>'의 위치를 찾는다. (찾으면 반복문 끝)
4. 결과값이 되는 리스트 word에 '<' 부터 '>'까지 태그를 넣어준다.
5. idx를 '>' 인덱스 값으로 바꿔줌

 

문자로 이루어진 단어가 끝났을때는 3가지 경우가 있다.

1. 다음이 공백인 경우
2. 문장의 끝인 경우
3. 다음이 괄호 '<'인 경우

단어인지 여부를 확인하고 역순으로 저장할때는 다음과 같은 알고리즘으로 작성하였다.

• 기본적으로 괄호가 아닌 문자일경우엔 w에 추가되도록 했다.
• 공백을 만날경우 지금까지 추가된 w값을 word 리스트에 넣어준다.
• idx가 문자열 S의 길이와 같아질 경우 while문이 끝나는데 문장 끝에 있는 단어를 word 리스트에 추가해주기 위해 while문이 끝난뒤에 w가 있다면 역순으로 바꾸어 word 리스트에 추가하도록 했다.
• 다음이 '<'인 단어는 괄호 '<'를 만났을때 처리되도록 하였다.

 

새로 알게된 점

문자열을 거꾸로 역순으로 바꿀때 : ''.join(reversed([문자열]))

 

 

 

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

17413번: 단어 뒤집기 2