[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()풀이 후기
스택이 아닌 방법으로 접근했다가 시간 초과 당하고 바로 스택으로 틀어서 풀 수 있었다.
문제를 풀기 전에 시간복잡도를 고려해보고 접근하는 습관을 길러야겠다.