백준 / 12919번 / A와 B 2 / python 파이썬

 

문제 : https://www.acmicpc.net/problem/12919

 

 

 

문제

수빈이는 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

 

 

---

나의 풀이

S = input()
T = input()

def dfs(T):
    if len(T) == len(S):
        if S == T:
            print(1)
            exit()
        else:
            return 0

    if T[-1] == 'A':
        dfs(T[:-1])

    if T[0] == 'B':
        dfs(T[::-1][:-1])

dfs(T)
print(0)

 

S에서 추가하며 비교하는 것보다 T에서 제거하며 비교하는 것이 더 수월할 것이라 판단했다. 처음엔 while문과 pop()을 사용하여 T에서 값을 제거하며 비교했으나 한 방향으로만 계산이 되어서 실패하였다. 따라서 DFS 재귀로 풀어야했다.

 

중요한 것은 문자열 뒤에 b를 붙이고 문자열을 뒤집으므로, 아래의 코드는 옳지 않다 !

if T[-1] == 'B':
    dfs(T[:-1][::-1])