[Python] 백준 12904번 - A와 B

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

12904번: A와 B

 

처음에는 백트래킹을 사용해서 S를 T로 만드는 모든 경우를 다 찾아보려고 했었다.

그런데 시간제한에 걸려서.. 다른사람들이 푼 방식을 참고했다.

 

S를 T로 만들어주는 대신, T가 S가 되는지만 확인해주면 되는 간단한 문제였다!

문제 조건이 S의 마지막에 매번 A나 B를 추가해주기 때문에, T의 가장 마지막 문자만 떼어주며 S가 나올 수 있는지 확인하면 된다.

 

코드는 다음과 같다.

s = input()
t = input()

while len(t) >= len(s):
    if len(t) == len(s):
        if t == s:
            print(1)
        else:
            print(0)
    
    if t[-1] == 'A':
        t = t[:len(t)-1]
    elif t[-1] == 'B':
        t = t[:len(t)-1]
        t = t[::-1]