알고리즘 풀이/백준

[백준 9935] 문자열 폭발

mhko411 2021. 4. 13. 20:51
728x90

문제

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

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

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

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

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

 

입력

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

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

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

 

출력

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


접근

- 스택에 문자를 하나씩 넣는다.

- 현재 넣은 문자가 폭발 문자열의 끝 문자와 같다면 

- 폭발 문자열의 길이만큼 스택의 끝에서부터 검사하여 같을 때

- 해당 문자를 스택에서 제거한다. 

 

구현

for문으로 탐색을 하면서 문자를 하나씩 스택에 넣는다.

이때 폭발 문자열인 pattern의 끝과 비교하여 같을 때 pattern의 길이만큼 스택의 문자열과 비교한다.

(여기서 음수 인덱스에서 스택의 크기보다 벗어나면 에러가 날 줄 알았지만 그렇지 않았다.)

만약 폭발 문자열과 같다면 stack에서 제거한다.

string = input()
pattern = input()
length = len(pattern)

stack = []
for char in string:
    stack.append(char)
    if char == pattern[-1] and ''.join(stack[-length:]) == pattern:
        del stack[-length:]

if not stack:
    answer = "FRULA"
else:
    answer = ''.join(stack)
print(answer)

'알고리즘 풀이 > 백준' 카테고리의 다른 글

[백준 11279] 최대 힙  (0) 2021.04.18
[백준 11497] 통나무 건너뛰기  (0) 2021.04.17
[백준 2304] 창고 다각형  (0) 2021.04.13
[백준 2841] 외계인의 기타연주  (0) 2021.04.12
[백준 2504] 괄호의 값  (0) 2021.04.12