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

문제

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

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

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

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

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

입력

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

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

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

출력

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

예제 입력 1

mirkovC4nizCC44
C4

예제 출력 1

mirkovniz

예제 입력 2

12ab112ab2ab
12ab

예제 출력 2

FRULA

시간 제한	메모리 제한
2 초 (추가 시간 없음)	128 MB

문제 접근

• [초기접근]

  • 임시 문자열을 두고 입력으로 주어진 문자열을 앞에서부터 문자 하나씩 임시문자열에 쌓아가며 임시문자열의 뒷부분이 폭발문자열과 동일한 경우 폭발 문자열에 해당하는 부분을 임시문자열에서 제거하는 방식으로 구현하면 되겠다 생각하였음.

    s = input()
    target = input()
    
    answer = ""
    for ch in s:
      answer += ch
    
      if len(answer) >= len(target):
          if not answer[-1] == target[-1]:
              continue
    
          if target == answer[len(answer) - len(target) :]:
              answer = answer[: len(answer) - len(target)]
    
    print(answer if answer else "FRULA")

  • 초기 접근으로 위와 같은 코드를 작성하였지만 시간초과 발생.

  • 문제가 되는 부분을 answer = answer[: len(answer) - len(target)]으로 추정함. → python의 문자열 슬라이싱의 시간복잡도의 경우 O(N)이므로 폭발 문자열이 아닌 임시 문자열의 앞부분을 슬라이싱하는 경우 최악의 경우 N이 1,000,000에 가까워지므로 시간초과가 발생할 수 있음. (target == answer[len(answer) - len(target) :]에서의 문자열 슬라이싱의 경우 폭발 문자열은 최대 36자이기에 크게 시간복잡도에 영향을 미치지 않는다.)

• [개선된 문제 풀이]

  • 폭발 문자열을 제거하는 경우 문자열을 슬라이싱하는 것이 아닌 스택을 두어 폭발문자열의 길이만큼 pop()을 반복하는 것으로 변경하여 폭발 문자열 제거하는데 반복되는 횟수를 최대 36번으로 제한할 수 있었다.

코드

s = input()
target = input()

stack = []
for ch in s:
    stack.append(ch)

    if len(stack) >= len(target):
        if not stack[-1] == target[-1]:
            continue

        if target == "".join(stack[len(stack) - len(target) :]):
            for _ in range(len(target)):
                stack.pop()

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