백준 9935번 문자열 폭발

정보

문제 바로가기 [클릭]

난이도: Gold4

관련 개념: #자료구조 #문자열 #스택

---

조건

시간 제한	메모리 제한
1 초	256 MB

---

문제

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

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

문자열이 폭발 문자열을 포함하고 있는 경우에, 모든 폭발 문자열이 폭발하게 된다. 남은 문자열을 순서대로 이어 붙여 새로운 문자열을 만든다. 새로 생긴 문자열에 폭발 문자열이 포함되어 있을 수도 있다. 폭발은 폭발 문자열이 문자열에 없을 때까지 계속된다. 상근이는 모든 폭발이 끝난 후에 어떤 문자열이 남는지 구해보려고 한다. 남아있는 문자가 없는 경우가 있다. 이때는 “FRULA”를 출력한다.

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

---

입력

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

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

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

---

출력

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

---

예제 입출력 1

입력

mirkovC4nizCC44
C4

출력

mirkovniz

---

예제 입출력 2

입력

12ab112ab2ab
12ab

출력

FRULA

---

코드(파이썬)

import sys


line = sys.stdin.readline().rstrip()
explosion = sys.stdin.readline().rstrip()
stack = list()

if len(explosion) > 1:
    line = list(line)
    
    for index, ch in enumerate(line):
        if ch == explosion[0]:
            stack.append([1, [index]])
        elif stack and ch == explosion[stack[-1][0]]:
            if ch == explosion[-1]:
                _, blank = stack.pop()
                
                line[index] = ''
                for j in blank:
                    line[j] = ''
                
            else:
                stack[-1][0] += 1
                stack[-1][1].append(index)
        else:
            stack = list()
            
    result = ''.join(line)
else:
    result = line.replace(explosion, '')

sys.stdout.write(result if result else "FRULA")

---

특이사항

• 다른 코드에 비해 3~4배 가량 메모리, 시간을 더욱 차지함
• 내 코드(177,928KB, 1,184ms)에 비해 우수한 코드(바로가기, 42,044KB, 292ms)와 비교

  • def main():
     
        string = input()    # 전체 문자열
        bomb = input()      # 폭발 문자열
        
        lastChar = bomb[-1] # 폭발 문자열의 마지막 글자
        stack = []
        length = len(bomb)  # 폭발 문자열의 길이
        
        for char in string:
            stack.append(char)
            if char == lastChar and ''.join(stack[-length:]) == bomb:
                del stack[-length:]
        
        answer = ''.join(stack)
        
        if answer == '':
            print("FRULA")
        else:
            print(answer)
        
        
    if __name__ == '__main__':
        main()
    
    

  • 스택에 저장하는 값
    • 기존: 다음 확인할 폭발 문자열 인덱스, 폭발 문자열과 일치하는 원래 문자열의 글자 인덱스
    • 개선: 원래 문자열의 모든 문자
  • 스택 초기화
    • 기존: 폭발 문자열과 일치하지 않을 때
    • 개선: 지금 글자가 폭발 문자열의 마지막 글자이면서 stack에 저장된 문자열이 폭발 문자열과 일치할 때

---

참고문헌

-