[BOJ] 9935. 문자열폭발 (python)

https://www.acmicpc.net/problem/9935

9935번: 문자열 폭발

 

문제

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

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

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

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

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

입력

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

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

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

출력

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

 

 

풀이

스택을 이용한 문제이다.
문자열을 index 0부터 읽어가며 stack에 넣고
넣은 후, 맨 뒤의 연결된 string이 폭발 문자열이라면 그만큼 pop 해준다.
그리고 다시 계속 확인한다!

 

코드

 

import sys
read = sys.stdin.readline

initial_string = read().strip()
bomb_string = read().strip()

bomb_len = len(bomb_string)

main_stack = []
bomb_point = 0

for string in initial_string:
    main_stack.append(string)
    if len(main_stack) >= bomb_len:
        if bomb_string == ''.join(main_stack[bomb_point:bomb_point + bomb_len]):
            for _ in range(bomb_len):
                main_stack.pop()
    bomb_point = len(main_stack) - bomb_len + 1
if main_stack:
    print(''.join(main_stack))
else:
    print('FRULA')