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

9935번: 문자열 폭발

시간 제한	메모리 제한	문제 티어
2 초(추가 시간 없음)	128 MB	골드 IV

📜 문제

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

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

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

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

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

 

📥입력

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

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

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

 

📤출력

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

 

💡풀이

 

나는 맨 처음에 파이썬의 내장함수인 replace를 사용하여 문제를 풀었다. 그러나 replace가 시간이 오래 걸릴 수 있다고 들었다. 실제로 채점해보면 예제는 맞으나 시간초과가 발생했다.

import sys

input = sys.stdin.readline

str = input().rstrip()
explode_str = input().rstrip()

while True:
  if str == '': 
    print('FRULA')
    break
  elif str.find(explode_str) != -1:
    str = str.replace(str[str.find(explode_str):str.find(explode_str) + len(explode_str)], '')
  else:
    print(str)
    break

 

따라서 나는 스택을 이용해서 문제를 풀었다.문자열에서 문자 하나씩 스택에 추가한 다음, 폭발 문자열과 일치하면 해당 문자열을 pop 시킨다.

import sys

input = sys.stdin.readline

# stack_queue.py 에 저장
class Stack:
	def __init__(self):
		self.items = []	# 데이터 저장을 위한 리스트 준비
	def push(self, val):
		self.items.append(val)
	def pop(self):
		try:	# pop할 아이템이 있으면
			return self.items.pop()
		except IndexError:	# pop할 아이템이 없으면 indexError 발생
			print("Stack is empty")
	def top(self):
		try:
			return self.items[-1]
		except IndexError:
			print("Stack is empty")
	def __len__(self):	# len()로 호출하면 stack의 item 수 반환
 		return len(self.items)
	def isEmpty(self):
		return self.__len__() == 0

str = input().rstrip()
explode_str = input().rstrip()

S = Stack()

for i in range(len(str)):
  S.push(str[i])
  if ''.join(S.items[-len(explode_str):]) == explode_str:
    for _ in range(len(explode_str)):
      S.pop()

if S:
  print(''.join(S.items))
else:
  print('FRULA')​

여기서 "Stack을 Python의 List를 활용하여 사용하면 되지 굳이 class를 만들어서 쓰는 것인가?" 라는 의문점이 생길 수 있다.

맞다. 실제로 list는 다양한 기능이 존재하고 그 중 몇 가지 기능을 활용해서 Stack을 구현할 수 있다. 그러나 Stack은 딱 두 가지 함수인 push와 pop으로만 내부 값을 변경할 수 있다. 즉, List를 Stack으로 직접 구혀내서 사용하면 append와 pop 이외에 다른 함수를 사용해 내부 값을 참고하거나 변경하게 되는 실수를 범할 수 있다. 이러한 실수를 줄이기 위해 애초에 제공되는 연산을 딱 한정해서 그 연산으로만 내부 값을 바꾸기 위해 사용되는 것이다!