백준 12904 파이썬 - A와 B

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

12904번: A와 B

문제

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

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

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

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

코드

import sys
s = list(sys.stdin.readline().rstrip())
t = list(sys.stdin.readline().rstrip())
flag = False
while t:
    if t[-1] == 'A':
        t.pop()
    elif t[-1] == 'B':
        t.pop()
        t.reverse()
    if s == t:
        flag = True
        break
sys.stdout.write('1' if flag else '0')

풀이

 s의 뒤에 문자열을 더하는 방식 보다는 t에서 빼가면서 확인하는 방식으로 while문을 돌고, flag가 True가 되면 1 아니면 0을 출력하면 된다.