백준 #9935

기억할 것

1. 스택 개념을 숙지했다고 해서 그것이 바로 적용되는 것은 아니다.

2. 그렇다면 스택을 어떻게 쓰는가?

스택을 안 써도 코드는 돌아가게 만들 수 있다. 알량하게 몇 문제 풀어보면서 느낀 거지만 스택, 큐, 덱을 안 쓰면 시간 초과가 난다. (물론 개념을 써도 추가적인 아이디어가 필요하여 시간초과가 나는 경우도 있긴 하지만 말이다..)

 

시도했지만 맞지 못한 문제에 한 몇달 묵히다가 알고리즘 공부한걸로 해결하니까 기분이 좋다. 

문제

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

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

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

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

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

입력

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

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

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

출력

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

예제 입력 1 복사

mirkovC4nizCC44
C4

예제 출력 1 복사

mirkovniz

예제 입력 2 복사

12ab112ab2ab
12ab

예제 출력 2 복사

FRULA

 

처음에 짠 무지성 코드

무작정 입력 받고 폭탄을 찾는 방식으로 재귀함수를 만들었다. 

#입력
import sys 
word = sys.stdin.readline().strip()
bomb = sys.stdin.readline().strip()

#재귀함수
def f(word):
    
    #기저상태
    if word == '':
        return 'FRULA'
    if bomb not in word:
        return word
    
    #재귀
    while True:
        location = word.find(bomb)
        word = word[:location] + word[location+len(bomb):]
        if bomb not in word:
            return f(word)
#출력
print(f(word))

 

정답 코드

어떤 느낌으로 짜야 하냐면

문자 하나 하나를 쌓다가 문자열의 축적이 우연히 폭탄이 되면 그때 터트리는 느낌..?

#입력
import sys 
word = sys.stdin.readline().strip()
bomb = sys.stdin.readline().strip()

#스택 
stack = []
for i in word:
    stack.append(i) #word의 단어를 스택에 하나씩 추가
    if ''.join(stack[len(stack)-len(bomb):]) ==bomb: #혹시 스택의 맨 마지막 단어들이 폭발할 단어면
        for i in range(len(bomb)):#폭발시킨다.
            stack.pop()  

#출력
if len(stack) == 0:
    print('FRULA')
else:
    print(''.join(stack))