[Baekjoon] 9935번 문자열 폭발

문제

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

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

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

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

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

입력

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

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

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

출력

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

예제

Example 1:

Input: 
mirkovC4nizCC44
C4
Output: 
mirkovniz

**Example2: **

Input:
12ab112ab2ab
12ab
Output:
FRULA

조건

시간 제한 : 2초

메모리 제한 : 128 MB

풀이과정

풀이

굉장히 많은 실패를 겪었던 문제.

처음에는 폭발시키면서 문자열을 그때마다 더해주었다. 시간 복잡도가 매우 커서 4%정도에 머물렀다.

다음으로는 문자열을 사용하였다. 시간복잡도가 조금은 줄었지만 48%에서 넘어가지 않았다. 하지만 문자열을 스택으로 바꾼 결과 해결이 되었다. 스택이 시간 복잡도가 더 좋은 듯하다.

그래도 한두달전에는 손도 못댔을거같은데 구현이 되었다는 점에 만족하고 좀 더 노력해야겠다.

import sys
input = sys.stdin.readline

# 입력 받기
s = input().rstrip()
target = input().rstrip()

# 마지막 문자 받기
lastt = target[-1]

# 폭발 문자열 길이 받기 / 스택 및 스택 길이 선언
n2 = len(target)
ans = []
len_ans = 0

# 하나씩 입력받으면서 계산
for i in s:
    if not ans:
        ans.append(i)
        len_ans += 1
    else:
        ans.append(i)
        len_ans += 1

        if i == lastt:
            # 폭발하는지 검사
            while(1):
                boom = False
                # 길이가 허용될때만
                if len_ans >= n2:
                    # 폭발한다면 폭발시키기
                    if "".join(ans[-n2:]) == target:
                        for _ in range(n2):
                            ans.pop()
                        len_ans -= n2
                        boom = True
                if boom == False:
                    break

# 정답 출력
if not ans:
    print("FRULA")
else:
    print("".join(ans))