[BOJ] 문자열 폭발 (#9935)

 

 

문제 

 

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

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

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

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

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

 

 

입력 

 

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

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

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

 

 

나의 코드

word = input()
bomb = input()

while True:
  if bomb in word:
    word = word.replace(bomb,"")
  if bomb not in word:
    break

if len(word) == 0:
  print("FRULA")
else:
  print(word)

분석 > 시간초과가 떴던 코드이다. bomb 길이가 백만으로 길어졌을 때를 고려하지 못하였다. 

 

효율적인 코드 

word = input()
bomb = list(input())
result = [] # 스택 만들기 

for i in range(len(word)):
  result.append(word[i]) # 문자를 이루는 원소들 result 배열에 추가
  if result[-len(bomb):] == bomb: # 추가되는 원소 중 bomb와 같으면
    for _ in range(len(bomb)): # bomb 길이만큼 뒤에서 pop
      result.pop()

if result:
  print(''.join(result))
else:
  print('FRULA')

분석 > 스택을 이용해서 푼 코드이다. 스택에 word의 각 문자를 순차적으로 append하다가 뒤에서부터 bomb의 길이만큼 슬라이싱한 문자가 bomb와 동일하다면 bomb의 길이만큼 pop하는 방법으로 풀었다. 스택을 이용해서 문제 푸는 것이 낯설었는데, 이번 문제를 통해 스택을 이용해서 문제 푸는 방법을 정확히 익힌 것 같다. 

 

 

9935번: 문자열 폭발