[백준] 9935: 문자열 폭발 python 구현 / 스택(Stack)

문제

상근이는 문자열에 폭발 문자열을 심어 놓았다. 폭발 문자열이 폭발하면 그 문자는 문자열에서 사라지며, 남은 문자열은 합쳐지게 된다.

폭발은 다음과 같은 과정으로 진행된다.

• 문자열이 폭발 문자열을 포함하고 있는 경우에, 모든 폭발 문자열이 폭발하게 된다. 남은 문자열을 순서대로 이어 붙여 새로운 문자열을 만든다.
• 새로 생긴 문자열에 폭발 문자열이 포함되어 있을 수도 있다.
• 폭발은 폭발 문자열이 문자열에 없을 때까지 계속된다.

상근이는 모든 폭발이 끝난 후에 어떤 문자열이 남는지 구해보려고 한다. 남아있는 문자가 없는 경우가 있다. 이때는 "FRULA"를 출력한다.

폭발 문자열은 같은 문자를 두 개 이상 포함하지 않는다.

풀이

처음에는 정규표현식(re)의 sub 메서드를 사용하여 문제를 해결하고자 했다.

그러나 시간초과 !

 

stack으로 풀어야 함을 인지하였다.

1. 문자열을 하나씩 결과값에 append
2. 폭발 문자열의 마지막과 같으면, 결과값의 마지막이 폭발 문자열과 같은지 확인하고
   • pop!

CODE

import sys

string = sys.stdin.readline().strip()
bomb = sys.stdin.readline().strip()

result = []

b_idx = 1
for i in string:
    result.append(i)
    if i == bomb[-1] and ''.join(result[-len(bomb):]) == bomb:
        for _ in range(len(bomb)):
            result.pop()
        
if result:
    print(''.join(result))
else:
    print('FRULA')