[Python 파이썬] 백준 9935번 - 문자열 폭발

2023년 11월 16일

문제 링크 : 백준 9935번 - 문자열 폭발

문제

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

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

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

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

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

입력

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

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

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

출력

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

알고리즘 분류

• 자료구조
• 문자열
• 스택

---

문제 접근

무식하게 문자열의 앞에서부터 폭발 문자열이 있으면 삭제하고, 다시 앞에서부터 비교해서 폭발 문자열이 있으면 삭제하고를 반복하면 시간초과가 발생하기 때문에 스택 자료구조를 활용하고자 하였다.

 

문자열의 앞에서부터 하나씩 스택에 append하며, 매번 스택의 맨 뒤에 폭발 문자열이 있는지 비교한다면 시간복잡도 O(n)으로 해결할 수 있으리라 가정했다.

 

---

순서도

 

1. 문자열과 폭발 문자열을 입력받는다.

2. 문자열의 맨 처음부터 하나씩 비교하며 쌓을 스택과 비교하고 있는 인덱스 번호를 저장할 변수 now를 선언한다.

3. 반복문 시작 -> 문자열 맨 마지막을 탐색할 때까지 반복

	3-1. 현재 탐색중인 문자열의 인덱스를 스택에 append한다.
    
    3-2. 스택에 append 할 때마다 현재 스택에서 폭발 문자열의 길이만큼 뒤에 있는 문자열들을 비교한다.
    
    3-3. 만약 같다면 폭발 문자열이므로 스택에서 pop해준다.
	
    3-4. 반복문을 한 번 수행할 때마다 now를 1만큼 증가시켜 스택에 append할 다음 문자열의 인덱스를 갱신시킨다.

4. 반복문 종료 후 스택이 비었다면 FLURA를, 아니라면 스택에 있는 최종 문자들을 조합해 결과 문자열을 출력한다.

---

입력 예제

 

mirkovC4nizCC44
C4

 

 

위 프로세스대로 진행한다면 다음과 같다.

 

1. 문자열의 맨 앞에서부터 차례대로 스택에 append 하며, 스택의 맨 뒤에서 2번째까지의 문자열이 'C4'인지 비교한다.

2. C4가 나올 때마다 폭발 문자열의 길이인 2만큼 pop해준다.

3. 최종 문자열을 출력한다. (만약 최종 스택이 비어있다면 FLURA를 출력)

 

출력 예제

 

mirkovniz

 

---

주의할 점

 

스택 자료구조로 접근하면 크게 어렵지 않은 문제라고 생각한다.

 

---

전체 코드

# 9935

def solution():
    stack = []
    now = 0
    bomb_len = len(bomb)
    str_len = len(str)

    while(now < str_len):
        stack.append(str[now])
        compair_str = ''.join(stack[-bomb_len : ])
        if compair_str == bomb:
            for _ in range(bomb_len):
                stack.pop()

        now += 1

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


if __name__ == "__main__":
    str = input()
    bomb = input()
    solution()

---

풀이 후기

 

스택이 아닌 방법으로 접근했다가 시간 초과 당하고 바로 스택으로 틀어서 풀 수 있었다.

 

문제를 풀기 전에 시간복잡도를 고려해보고 접근하는 습관을 길러야겠다.