백준 9935 문자열 폭발 (파이썬)

문자열, 스택, Gold 4

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

문제

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

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

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

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

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

입력

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

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

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

출력

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

첫번째 제출 코드 (시간초과)

word = input()
boom = input()

while boom in word:
    if boom in word:
        word = word.replace(boom, "", word.count(boom))
if word == "":
    print("FRULA")
print(word)

매번 문자열 안에 폭발 문자열이 있는지 확인하고 매번 repace로 치환하면서 풀어 시간초과가 발생했습니다.

아이디어는 맞았지만 효율성이 많이 부족한 것 같습니다. 그리하여 그런데 아이디어가 생각나지 않아 알고리즘만 보고 아이디어를 얻었습니다.

 

두번째 제출 코드 (정답)

word = input()
boom = input()
stack = []
last = boom[-1] #폭발문자열의 마지막
n = len(boom) #폭발문자열의 길이
for s in word:
    stack.append(s)
    if s == boom[-1]:
        if ''.join(stack[-1*n:]) == boom:
            for i in range(n):
                stack.pop()

if not stack:
    print("FRULA")
print(''.join(stack))

백준에서 사용된 알고리즘이 스택이라는 것을 보고 문자열의 앞에서부터 stack에 넣으면서 stack의 마지막이 boom의 마지막과 같을때 stack의 마지막부터 폭발 문자열의 길이만큼 비교하여 같으면 폭발 문자열의 개수만큼 pop을 해주어 문자를 제거했습니다. 

 

결과 및 정리

문자열 다루는 훈련을 하기 위해 문자열 위주로 풀고있는데 골드문제라서 단순 문자열만 다루지 않고 자료구조가 섞여있어 어려운 문제였습니다.

위 문제를 보고 스택을 떠올리는 것은 쉽지 않은 것 같고 지금 골드2 이긴 하지만 위 문제도 힘들어하는 것을 보니 아직 더 노력해야 할 것 같습니다.

 

추가) 리스트에서 삭제해야 될 길이와 위치를 안다면 pop 말고 del 함수가 더 좋은 것 같습니다.

if s == boom[-1]:
    if ''.join(stack[-n:]) == boom:
        del stack[-n:]