[백준] #12919 A와 B 2(python)

📌문제

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

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

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

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

입력

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

출력

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

예제 입력1

A
BABA

예제 출력1

1

예제 입력2

BAAAAABAA
BAABAAAAAB

예제 출력2

1

예제 입력3

A
ABBA

예제 출력3

0

백준 12919번 문제

💡Idea

[백준] 12904번 문제 A와 B 와 비슷한 문제입니다.

A와 B 문제 처럼 S ➡ T로 가는 과정을 추적하며 문제를 푸는 것이 아니라, 반대로 T ➡ S로 가면서 문제를 풀어나가면 됩니다.

하지만 A와 B 문제의 경우 아래와 같은 두 조건 중 무조건 하나만 충족할 수 있었습니다.

1. 문자열의 맨 뒤 문자가 A 인 경우
2. 문자열의 맨 뒤 문자가 B인 경우

따라서, 문자열의 뒤가 A인지 B인지에 따라서 만들어지는 문자열이 하나였던 반면에, 이 문제는 아래와 같이 두 가지 조건에 따라 여러 개의 문자열이 만들어질 수 있습니다.

1. 문자열의 맨 뒤의 문자가 A인 경우
2. 문자열의 맨 앞의 문자가 B인 경우
   $\;$
   ➡ BABA의 경우, 1번과 2번 조건을 모두 충족하기 때문에 1번을 적용하여 BAB가 만들어질 수도, 2번을 적용하여 ABA가 만들어질 수도 있다.

따라서, 만들어 질 수 있는 모든 문자열을 탐색할 수 있도록 BFS 혹은 DFS를 함께 사용해서 문제를 풀어야 합니다.

💻코드

DFS 코드(재귀함수 사용)

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

import sys

input = sys.stdin.readline

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

def dfs(cur_str):
    if len(cur_str) < len(S): return 0
    if cur_str == S: return 1
    if cur_str[-1] == 'A' and dfs(cur_str[:-1]): return 1
    if cur_str[0] == 'B'and dfs(cur_str[::-1][:-1]): return 1
    return 0

print(dfs(T))

BFS 코드(큐 사용)

• ⏰ 시간 : 92 ms / 메모리 : 32424 KB

import sys
from collections import deque

input = sys.stdin.readline

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

deq = deque([T])

while deq:
    cur_str = deq.popleft()
    if len(cur_str) < len(S): continue
    if cur_str == S:
        print(1)
        exit()
    if cur_str[-1] == 'A': deq.append(cur_str[:-1])
    if cur_str[0] == 'B': deq.append(cur_str[::-1][:-1])

print(0)