[BOJ] 백준 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을 출력한다.

 

https://www.acmicpc.net/problem/12904

12904번: A와 B

 

💡 풀이 및 코드

s = input()
t = input()

s_len, t_len = len(s), len(t)
for _ in range(t_len-s_len):
    if t[-1] == 'A':
        t = t[:-1]
    elif t[-1] == 'B':
        t = t[:-1]
        t = t[::-1]
if s == t:
    print(1)
else:
    print(0)

생각보다 풀이를 쉽게 떠올릴 수 있었다.

이 문제의 핵심은 S를 이용해서 T를 만들 수 있는 지가 아니라 T를 이용해서 S를 만들 수 있는지 판단하면 된다. S에서 'A'를 추가하거나 S를 뒤집은 후 'B'를 추가하는 연산만 가능하므로 반대로 T의 맨 지막 문자가 'A'라면 'A'를 삭제하고, 마지막이 'B'라면 'B'를 삭제한 후 문자열을 뒤집는다. 이렇게 만들어진 T와 S의 길이가 같아질 때 두 문자열이 같은지 다른지를 판별할 수 있다.