[Python] 백준 9935 문자열 폭발

문제

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

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

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

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

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

입력

첫째 줄에 문자열이 주어진다. 문자열의 길이는 1보다 크거나 같고, 1,000,000보다 작거나 같다.

둘째 줄에 폭발 문자열이 주어진다. 길이는 1보다 크거나 같고, 36보다 작거나 같다.

두 문자열은 모두 알파벳 소문자와 대문자, 숫자 0, 1, ..., 9로만 이루어져 있다.

출력

첫째 줄에 모든 폭발이 끝난 후 남은 문자열을 출력한다.

9935번: 문자열 폭발

---

접근방법 (시간초과)

1) 폭탄 문자열을 가진 구간을 좌우로 split

2) 리스트를 다시 join

3) 폭탄 문자가 없어질 때까지 반복한다. 

하지만, CCCCCCCCCC444444444444 이런 예시가 들어오면 O(n2)이 되어 시간 에러가 난다.

# 시간초과
import sys

word = sys.stdin.readline().rstrip()
x = sys.stdin.readline().rstrip()

while True:
    init = word
    tmp = word.split(x)
    word = ''.join(tmp)
    if len(tmp) == 1:
        if tmp[0] != "" :
            break
        elif tmp[0] == "" :
            word = 'FRULA'
            break

print(word)

접근방법 (통과)

1) For문: 문자열 차례로 Stack에 저장
2) stack의 마지막 글자가 같다면 (# 문자가 지워지고 재결합된 문자열을 바로 검증할 수 있음)

3) stack[-length(x): ] == 폭탄 문자열과 같으면 
4) For문: stack[-length(x): ]를 모두 pop!!

 

시간 복잡도는 O(n)이 된다. 

import sys

word = list(sys.stdin.readline().rstrip())
x = list(sys.stdin.readline().rstrip())

stack = []
for i in word:
    stack.append(i)
    if stack[-1] == x[-1] and len(stack) >= len(x): # 폭발문자열의 마지막과 Word의 i+1번쨰의 문자열 비교시 같을때
        if stack[-len(x):] == x:
            for i in range(len(x)): stack.pop()
            
if stack:
    print("".join(stack))
else:
    print("FRULA")