[백준] #12904 A와 B(python)

📌문제

수빈이는 A와 B로만 이루어진 영어 단어가 존재한다는 사실에 놀랐다. 대표적인 예로 AB (Abdominal의 약자), BAA (양의 울음 소리), AA (용암의 종류), ABBA (스웨덴 팝 그룹)이 있다.

이런 사실에 놀란 수빈이는 간단한 게임을 만들기로 했다. 두 문자열 S와 T가 주어졌을 때, S를 T로 바꾸는 게임이다. 문자열을 바꿀 때는 다음과 같은 두 가지 연산만 가능하다.

• 문자열의 뒤에 A를 추가한다.
• 문자열을 뒤집고 뒤에 B를 추가한다.

주어진 조건을 이용해서 S를 T로 만들 수 있는지 없는지 알아내는 프로그램을 작성하시오.

입력

첫째 줄에 S가 둘째 줄에 T가 주어진다. (1 ≤ S의 길이 ≤ 999, 2 ≤ T의 길이 ≤ 1000, S의 길이 < T의 길이)

출력

S를 T로 바꿀 수 있으면 1을 없으면 0을 출력한다.

예제 입력1

B
ABBA

예제 출력1

1

예제 입력2

AB
ABB

예제 출력2

0

백준 12904번 문제

💡Idea

문자열 S를 두 규칙을 적용하여 문자열 T로 만들어낼 수 있는지를 찾는 문제입니다.

처음에는 S에서 시작하여 두 규칙을 적용하여 만들 수 있는 길이 len(T)의 모든 문자열을 BFS를 통해 만들어 그 중에 T가 있는지를 확인해보았습니다.

하지만, 그런 식으로 풀면 메모리 초과가 발생합니다.

👇 메모리 초과가 발생한 BFS 코드

import sys
from collections import deque

input = sys.stdin.readline

S = input().strip()
T = input().strip()

deq = deque([S])
ans = 0
while deq:
    current_str = deq.popleft()
    if current_str == T:
        ans = 1
        break
    if len(current_str) >= len(T): continue
    deq.append(current_str + 'A')
    deq.append(current_str[::-1] + 'B')

print(ans)

문자열 T에서 거꾸로 문자열 S를 만들 수 있는지 확인하자!

이런 경우, 반대로 생각해서 문자열 T에서 문자열 S를 만들 수 있는지를 확인해보면 메모리와 시간을 훨씬 줄일 수 있다는 것을 알게 되었습니다.

따라서, T에서 규칙을 거꾸로 적용하여 S가 나오는지를 확인하여 문제를 풀어보았습니다!

💻코드

• ⏰ 시간 : 68 ms / 메모리 : 30840 KB

import sys
input = sys.stdin.readline

S = input().strip()
T = input().strip()

while len(T) > len(S):
    if T[-1] == 'A': T = T[:-1]
    else:
        T = T[:-1]
        T = T[::-1]

print(1 if S == T else 0)