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

 

 

두 개의 문자열이 주어진다.

첫 문자열에서 두 번째 문자열이 포함되면 그 문자열을 제거하고 남은 부분이 이어 붙인다.

이어 붙여진 문자열에 두 번째 문자열이 포함되어 있다면 위 과정을 계속 반복한다.

 

남은 문자열이 존재하면 그 문자열을 출력하고, 없다면 "FRULA"를 출력한다.

 

 

 

문제

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

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

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

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

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

입력

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

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

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

출력

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

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

 

폭탄 문자열의 마지막 철자$($=x$)$를 따로 저장해 둔다.

첫 문자열의 알파벳을 하나씩 확인하면서 스택에 저장한다.

만약 알파벳이 x와 같다면 스택의 뒷부분 중 폭탄 문자열의 길이만큼이 폭탄 문자열과 같은 지 확인한다.

그래서 같다면 폭탄 문자열을 스택에서 제거한다.

 

import sys
input = sys.stdin.readline
print = sys.stdout.write

def main():
    string = input().rstrip()
    bomb = list(input().rstrip())
    b = len(bomb)
    x = bomb[-1]

    stack = []
    for s in string:
        stack.append(s)
        if s == x and stack[-b:] == bomb:
            del stack[-b:]

    print("".join(stack) if stack else "FRULA")

if __name__ == "__main__":
    main()

 

 

 

예제 입력 1 

mirkovC4nizCC44
C4

예제 출력 1 

mirkovniz

예제 입력 2 

12ab112ab2ab
12ab

예제 출력 2 

FRULA