[ 17413 ] [ 스택 ] 단어 뒤집기 2

 

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

17413번: 단어 뒤집기 2

 

---

풀이 과정

---

• 이전 문제랑 크게 다르지 않은듯 하면서 많이 다른 문제.. split을 쓰는것보다는 직접 진행하는것이 훨씬 쉽게 풀었던 것 같음.. 이전 문제가 더 쉬웠는데 왜 이번 문제가 정답률이 높지??..

 

1. 왼쪽에서 진행하면서 '<' 문자를 만났을 때와 안만났을 때로 구분한다.
   • '<' 문자를 만났을 때
     • 이전 스택에 저장되어있던 문자를 모두 pop해주면서 저장 (이전 문자는 뒤집힘)
     • 문자를 그대로 저장
   • '<' 문자를 만나지 않았을 때
     • 문자를 스택에 저장
2. 공백을 만나게 되면(' ')
   • '<>' 내부라면 그냥 저장해주면 된다.
   • '<>' 내부가 아니라면 공백을 만나기 전까지 스택에 저장된 문자들을 모두 pop해주면서 저장하고, 공백을 뒤에 붙여준다.
3. 맨 마지막에, 스택에 문자가 남아있다면 모두 pop해준 다음 저장해둔다.
4. 저장해둔 문자를 출력한다.

---

소스 코드

---

import sys

input = lambda : sys.stdin.readline().rstrip()

sentence = input()
stack = []
skip = False
answer = ''
for ch in sentence:
    if ch == '<':
        while stack:
            answer += stack.pop()
        skip = True
    elif ch == '>':
        answer += '>'
        skip = False
        continue

    if skip:
        answer += ch
    else:
        if ch == ' ':
            while stack:
                answer += stack.pop()
            answer += ch
        else:
            stack.append(ch)

while stack:
    answer += stack.pop()

print(answer)