Tiny Bunny
본문 바로가기
Algorithm/Python

[BOJ #9935] 문자열 폭발

by nowag 2024. 5. 14.

문제

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

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

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

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

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

 

입력

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

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

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

 

출력

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

 

내가 쓴 코드1 (틀 -> 시간초과)

# 시간초과.
string = input()
bomb = input()
b = len(bomb)

idx = string.find(bomb)
while idx >= 0:
    string = list(string)
    for i in range(b):
        string.pop(idx)
    string = ''.join(string)
    idx = string.find(bomb)

if not string:
    print('FRULA')
else:
    print(string)

 

내가 쓴 코드2 (정답!)

string = input()
bomb = input()
b = len(bomb)
ans = []
for s in string:
    ans.append(s)
    if ''.join(ans[-b:]) == bomb:
        for _ in range(b):
            ans.pop()

if not ans:
    print('FRULA')
else:
    print(''.join(ans))

처음에 풀때 설마 이거 뒤에서부터 하는거겠어.. 하고 find() 써서 했는데,, 처참히 실패,, 진짜 설마 했던 뒤에서부터 하는 게 맞았던 것이었다.... 근데 슬라이싱할 생각은 못했었다,, 쉬운데 어렵고 어려운데 쉬워;;

 

제출 결과

728x90
반응형

'Algorithm > Python' 카테고리의 다른 글

[BOJ #7576] 토마토  (0) 2024.05.15
[BOJ #14940] 쉬운 최단거리  (1) 2024.05.15
[BOJ #9012] 괄호  (0) 2024.05.13
[BOJ #10026] 적록색약  (0) 2024.05.09
[BOJ #3187] 양치기 꿍  (0) 2024.05.07