[백준 12919번] 파이썬 - A와 B 2

백준 12919_A와 B 2

시간 제한 2초, 메모리 제한 512MB

# 조건

• 수빈이는 A와 B로만 이루어진 영어 단어 존재한다는 사실에 놀랐다.
• 대표적인 예로 AB (Abdominal의 약자), BAA (양의 울음 소리), AA (용암의 종류), ABBA (스웨덴 팝 그룹)이 있다.
• 이런 사실에 놀란 수빈이는 간단한 게임을 만들기로 했다.
• 두 문자열 S와 T가 주어졌을 때, S를 T로 바꾸는 게임이다.
• 문자열을 바꿀 때는 다음과 같은 두 가지 연산만 가능하다.
  • 문자열의 뒤에 A를 추가한다.
  • 문자열의 뒤에 B를 추가하고 문자열을 뒤집는다.
• 주어진 조건을 이용해서 S를 T로 만들 수 있는지 없는지 알아내는 프로그램을 작성하시오.

입력

• 첫째 줄에 S가 둘째 줄에 T가 주어진다. (1 ≤ S의 길이 ≤ 49, 2 ≤ T의 길이 ≤ 50, S의 길이 < T의 길이)

출력

• S를 T로 바꿀 수 있으면 1을 없으면 0을 출력한다.

# 접근 방법

• 할 수 있는 행동은 나와있는 대로 2가지이다.
• S에서 T를 만들기 위해서는 모든 경우의 수를 봐야 하지만
• 반대로 T에서 S를 만드는 것은 가장 뒤에 있는 글자가 무엇인지에 따라서 할 수 있는 행동이 정해지므로 더 쉬워보인다.
• 따라서 기존에 나와있는 규칙 2개를 역으로 실행해준다.
  • 제일 뒤에 A가 있다면 A를 제거한다.
  • 문자열 제일 앞이 B라면 뒤집고 B를 제거해준다.
  • 제일 뒤가 A이고 제일 앞이 B이면 2가지의 행동을 재귀로 탐색해준다.
• 길이가 같은 경우 S와 T가 같지 않다면 0을 리턴해주고 같다면 1을 리턴해준다.

import sys  
sys.stdin = open('input.txt')  
input = sys.stdin.readline  

def check(a, b):  
    if len(a) == len(b):  
        if a != b:  
            return 0  
        else:  
            return 1  

    # 제일 뒤가 A이고 제일 앞이 B이면 재귀로 2가지 경우 모두 탐색  
    if b[-1] == 'A' and b[0] == 'B':  
        val1 = check(a, b[:-1])  
        val2 = check(a, b[:0:-1])  
        if val1 or val2:  
            return 1  

    elif b[-1] == 'A':  
        val3 = check(a, b[:-1])  
        if val3:  
            return 1  

    elif b[0] == 'B':  
        val4 = check(a, b[:0:-1])  
        if val4:  
            return 1  
    else:  
        return 0  

    return 0  

S = list(input().rstrip())  
T = list(input().rstrip())  



answer = check(S, T)  
print(answer)