[🥈3 / 백준 17413 / 파이썬] 단어 뒤집기 2

 

17413번: 단어 뒤집기 2

 

문제

문자열 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>
예제 입력 4 one1 two2 three3 4fourr 5five 6six	예제 출력 4 1eno 2owt 3eerht rruof4 evif5 xis6
예제 입력 5 <int><max>2147483647<long long><max>9223372036854775807	예제 출력 5 <int><max>7463847412<long long><max>7085774586302733229
예제 입력 6 <problem>17413<is hardest>problem ever<end>	예제 출력 6 <problem>31471<is hardest>melborp reve<end>
예제 입력 7 <   space   >space space space<    spa   c e>	예제 출력 7 <   space   >ecaps ecaps ecaps<    spa   c e>

 

풀이

 

스택을 사용해도 되고, 문자열 슬라이싱 등 다른 방법을 사용해도 풀 수 있는 문제이다.

본인은 스택을 사용하여 풀었으므로 스택을 사용한 코드를 기재한다.

 

<, > 안에 있는 문자열은 그대로 출력하고, 그 밖에 있는 문자들을 스택에 담는다.

공백이 있을 경우 스택에 담았던 문자열을 뒤집어 출력한다.

 

ans = ""
tag = False
stack = ""
for i in input():
    if i == "<":
        tag = True
        ans += stack[::-1]
        stack = ""
        ans += i
        continue
    elif i == ">":
        tag = False
        ans += i
        continue
    elif i == " ":
        ans += stack[::-1] + " "
        stack = ""
        continue
        
    if tag:
        ans += i
    else:
        stack += i
print(ans+stack[::-1])