[python]백준 12904 풀이

A와 B

백준 12904 문제 링크

수빈이는 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을 출력한다.

풀이

이 문제는 단순하게 생각해낸 풀이를 코드로 작성하면 쉽게 해결할 수 있는 문제입니다. 우선은 S와 T는 조건에 의하면 처음부터 무조건 같을수는 없습니다. 따라서 무조건 다른 글자입니다.

기본적으로 해당 연산을 결과로부터 거꾸로 수행하면 쉽게 찾아낼 수 있겠죠? 왜냐하면 연산 종류에 따라서 뒷 글자가 결정되니까요. T에 뒷글자를 확인해서 해당 문자가 A라면 단순하게 문자를 제거한 후 S와 비교하고, B라면 제거한 후 뒤집어서 S와 비교하면 되겠죠.

import sys
input = sys.stdin.readline

s = input().rstrip()
t = list(input().rstrip())
for i in range(len(t)):
	# 뒷글자가 A인 경우
    if t[-1] == 'A':
        t.pop()
    # 뒷글자가 B인 경우
    else:
        t.pop()
        t.reverse()
    
    # 연산 후 
    if ''.join(t) == s:
        print(1)
        exit()
print(0)

만약 같다면 1을 출력하고 프로그램을 종료하면 됩니다. 그렇지 않은 경우에는 0을 출력시키겠습니다.