[백준] #9935 문자열 폭발(python)

수영·2022년 8월 25일

Python3 백준 스택

백준

목록 보기

49/117

📌문제

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

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

문자열이 폭발 문자열을 포함하고 있는 경우에, 모든 폭발 문자열이 폭발하게 된다. 남은 문자열을 순서대로 이어 붙여 새로운 문자열을 만든다.
새로 생긴 문자열에 폭발 문자열이 포함되어 있을 수도 있다.
폭발은 폭발 문자열이 문자열에 없을 때까지 계속된다.
상근이는 모든 폭발이 끝난 후에 어떤 문자열이 남는지 구해보려고 한다. 남아있는 문자가 없는 경우가 있다. 이때는 "FRULA"를 출력한다.

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

입력

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

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

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

출력

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

예제 입력1

mirkovC4nizCC44
C4

예제 출력1

mirkovniz

예제 입력2

12ab112ab2ab
12ab

예제 출력2

FRULA

백준 9935번 문제

💡Idea

문제를 보면 replace를 반복하면 해결할 수 있을 것처럼 보이지만, 그렇게 호락호락한 문제가 아닐 것 같다는 생각이 듭니다.

실제로 아래와 같이 replace를 사용하면 시간 초과가 발생합니다.

import sys
input = sys.stdin.readline
str = input().strip() # 입력 문자열
explosion_str = input().strip() # 폭발 문자열
prev = '' # replace 이전의 문자열 저장
while prev != str: # replace를 더 이상 할 게 없다면 종료
    prev = str
    str = str.replace(explosion_str, '')

print(str if str else 'FRULA')

replace의 경우, 시간복잡도는
$O($ 문자열의 길이 $\times$ (교체할 문자열의 길이 + 교체되는 문자열의 길이) $)$
입니다.
따라서 위의 코드는 최악의 경우 $O(N^2)$ 의 시간복잡도를 가질 수 있습니다.

따라서 이 문제는 stack의 pop을 활용하여 시간 초과 없이 해결할 수 있습니다.

💻코드

⏰ 시간 : 584 ms / 메모리 : 42020 KB

import sys
input = sys.stdin.readline
str = input().strip() # 입력 문자열
explosion_str = list(s for s in input().strip()) # 폭발 문자열
stack = [] # 스택
for s in str:
    stack.append(s)
    if stack[-1] == explosion_str[-1] and stack[-len(explosion_str):] == explosion_str:
        for _ in range(len(explosion_str)):
            stack.pop()
print(''.join(stack) if len(stack) else 'FRULA')