[🥇4 / 백준 9935 / 파이썬] 문자열 폭발

 

9935번: 문자열 폭발

 

문제

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

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

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

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

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

입력

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

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

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

출력

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

 

예제 입력 1  mirkovC4nizCC44 C4	예제 출력 1  mirkovniz
예제 입력 2  12ab112ab2ab 12ab	예제 출력 2  FRULA

 

풀이

 

입력받은 문자열에 특정 문자열이 있는 지 확인하여 지워나가는 방식의 문제이다.

 

단순히 문자열에 폭발 문자열이 없을 때까지 순회하며 문자열을 탐색하는 방법은

시간복잡도가 O(N²)가 될 수 있어 시간초과가 뜰 것이다.

 

따라서 위 문제를 효율적으로 처리할 수 있는 스택을 사용하여 문제를 해결했다.

 

 

우선 처음 입력받은 문자열을 슬라이싱하여 하나씩 for문으로 전달받는다.

 

그리고 이를 스택에 넣고, 만약 스택의 크기가 폭발 문자열의 길이보다 길어진다면

스택의 끝부분을 폭발 문자열의 길이만큼 검사하여 서로가 같은지 체크한다.

 

만약 서로가 같다면, 폭발 문자열의 길이만큼 pop해준다.

 

import sys
input = sys.stdin.readline

s = input().rstrip()
# 비교를 쉽게 하기 위해 폭발 문자열을 리스트로 변형했다.
b = list(input().rstrip())
q = []

for i in s:
    q.append(i)
    if len(q) >= (len(b)):
        if q[-len(b):] == b:
            for _ in range(len(b)):
                q.pop()
print(*q if q else "FRULA",sep="")