(백준) 12904 파이썬 - A와 B

(백준) 12904 - A와 B

문제

수빈이는 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 = input()
T = input()
is_possible = 0

def fun(word, goal):
    if len(word) > len(goal):
        return
    if word == goal:
        global is_possible
        is_possible = 1
        return
    
    fun(word+"A", goal)
    fun("".join(reversed(word))+"B", goal)

fun(S, T)

print(is_possible)

 

 

결과는 당연하게 시간 초과가 떴고 조금 생각을 바꿔서 T에서 역연산을 해서 S를 만들 수 있는지 계산하게 했다.

 

 

import sys

S, T = [list(sys.stdin.readline().strip()) for _ in range(2)]

while len(S) != len(T):
    if T[-1] == 'A':
        T.pop()
    else:
        T.pop()
        T = T[::-1] # 문자열 뒤집기

if S == T:
    print(1)
else:
    print(0)