[solved.ac 골드4] 9935_문자열 폭발 (파이썬, 문자열, 스택)

제한 사항

문제

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

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

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

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

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

입력

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

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

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

출력

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

예제

---

나의 풀이

매번 폭발 문자열을 찾아서 제거하는 과정을 반복하는 것은 상당한 시간이 소요될 것이다. (그게 됐다면 이 문제가 골드4 난이도일 이유도 없겠지)

어떻게 풀까 고민하다가 이 문제의 유형에 스택이 적혀있는 것을 보고 힌트를 얻어서 풀 수 있었다.

원래 문자열을 한 글자씩 스택에 넣다가 마지막으로 스택에 들어간 문자가 폭발 문자열의 마지막 문자와 같다면 그때서야 스택의 마지막 부분에서 폭발 문자열의 길이만큼 폭발 문자열과 비교하고 일치한다면 제거하는 것이다.

스택을 사용했을 때 장점은 이미 통과된 앞에서부터 다시 확인하지 않아도 되고 폭발 문자열이 제거된 위치부터 다시 확인할 수 있기 때문에 비효율적인 비교 작업을 하지 않아도 된다는 점이다.

위 과정으로 원래 문자열을 모두 스택에 넣고 나면 폭발 문자열이 모두 제거된 문자들을 얻을 수 있다. -> 스택에 쌓여있다.

만약 스택에 문자가 들어있다면 문자열로 묶어서 출력해주면 되고, 스택이 비었다면 "FRULA"를 출력하면 된다.

s, bomb = input(), input()
bl = len(bomb)
stack = []

for c in s:
    stack.append(c)
    if len(stack) >= bl and c == bomb[-1] and ''.join(stack[-bl:]) == bomb:
        del stack[-bl:]

print(''.join(stack)) if stack else print('FRULA')

채점 결과

(두 풀이는 완전히 동일하고 PyPy3와 Python3의 차이만 있다)