백준 12094: A와 B (Python)

문제

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

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

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

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

해설

문제 설명이 단순한 것에 비해 발상이 조금 까다롭게 느껴진 문제다.

S를 T로 바꿀 때, 두 가지 연산이 가능하다.

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

두 연산 모두 '문자열 뒤에' 새로운 문자를 추가함을 알 수 있다.

그러면 우리는 최종적으로 만들어야 하는 문자 T에 대해, '마지막으로 추가된 문자'가 무엇인지를 기준으로 단어를 역추적할 수 있다.

 

다음 예제를 보며 이해해보자

# 예제 입력 1
B
ABBA

 

이제, T에 대해 마지막 문자를 기준으로 역추적해보자.

 

1) ABBA

>> 마지막 문자가 'A' 이므로, 이전 문자는 'ABB' 였을 것이다.

2) ABB

>> 마지막 문자가 'B' 이므로, 이전 문자는 'BA' 였을 것이다.

3) BA

>> 마지막 문자가 'A' 이므로, 이전 문자는 'B' 였을 것이다.

 

그러므로 ABBA 를 만드는게 가능한 문자는 ['B', 'BA', 'ABB'] 뿐이라는 이야기가 된다.

따라서 위 예제 입력의 정답은 1이 된다.

 

이 아이디어를 토대로 구현만 하면 된다.

코드

S = input()
T = input()

stack = list(T)
result = 0
while stack:
    if stack == list(S):
        result = 1
        break
        
    last_ch = stack.pop()

    if last_ch == 'B':
        stack = list(reversed(stack))
        
print(result)