17413 단어 뒤집기2 _python

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

17413번: 단어 뒤집기 2

스택문제라고 하지만 기본문자열을 다루는것으로 간단하게 풀린다.

 

문제 접근

문자열을 "<"와 ">", " "이렇게 3가지로 구분해서 경우에 따라 상황을 적용하면 된다.

그런데 "<"가 나오면 ">"가 나올 때까지 뒤집지않으므로 구분해주는 방법을 추가해주면 된다.

 

코드

import sys
input = sys.stdin.readline

#"<"를 만나면 ">"를 만날 때 까지 그냥 지나감.
#나머지는 [n:m]으로 끄집어와서 reversed로 뒤집음.
#n은 ">"를 만나거나 " "를 만나면 시작.

S = list(input().rstrip())

start = 0
count = 0

def reverseS():
    temp = S[start:i]
    temp.reverse()
    S[start:i] = temp
i=0
while i < len(S):
    if S[i] == "<":
        reverseS()
        while S[i] != ">":
            i += 1
        start = i+1
    elif S[i] == " ":
        reverseS()
        start = i+1
        i += 1
    else:
        i += 1

if S[-1] != ">":
    reverseS()

print("".join(S))

스택을 쓰지않았기 때문에 "<"가 나오면 ">"가 나올때까지 넘어가는 방식으로 풀었다.

 

스택을 사용한다면 뒤집어야 하는부분은 스택에 담아두고, "<"부터">"를 구분하기위해 bool변수를 하나 사용한다.

import sys
input = sys.stdin.readline

S = list(input().rstrip())

ans = ""
stack = ""
tag=False
for i in S:
    if i=="<":
        tag = True
        ans += stack[::-1]
        stack=""
        ans += i
    
    elif i==">":
        tag = False
        ans += i

    elif i==" ":
        ans += stack[::-1]
        ans += " "
        stack = ""

    else:
        if tag:
            ans+= i
        else:
            stack += i

print(ans + stack[::-1])

생각한것보다 코드에 손가는부분이 많아지더라