[백준 / Python] 9935번 문자열 폭발

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

9935번: 문자열 폭발

문제 분석

난이도

골드4

분류

스택

들어가기 전에

대략적인 시간복잡도 계산을 할 수 있다면 어려울 것이 없는 문제.

문제

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

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

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

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

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

입력

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

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

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

출력

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

 

문제 풀이

 

풀이

단순히 구현을 해주면 된다.

시간복잡도를 계산해보면 100만개의 글자를 넣으면서 검사해야되는 길이의 최대 길이만큼 계속 검색해도(36)

36*1,000,000 = 36,000,000 겨우 3600만밖에 안되므로 최적화나 그런 것에 신경쓸 필요가 없다.

스택 안의 내용물이 폭발 문자열보다 작은지 등 몇가지만 신경써서 구현해주면 된다.

소스코드

from sys import stdin
input=stdin.readline

s1=input().rstrip()
s2=input().rstrip()
stack=[]

idx=0
while len(stack)<len(s2):
    stack.append(s1[idx])
    idx+=1

while idx<len(s1):
    if len(stack)>=len(s2):
        able=True
        for i in range(len(s2)):
            if stack[-len(s2)+i]!=s2[i]:
                able=False
                break
        if able:
            for i in range(len(s2)):
                stack.pop()
    stack.append(s1[idx])
    idx+=1

able=True
for i in range(len(s2)):
    if stack[-len(s2)+i]!=s2[i]:
        able=False
        break
if able:
    for i in range(len(s2)):
        stack.pop()
print(''.join(stack) if stack else 'FRULA')

후기

스택 연습용으로 적당한 문제라고 생각한다.