[BaekJoon] 12919 A와 B 2 (Gold 5) - Python

문제

 

수빈이는 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을 출력한다.

 

입력예시1)

A
BABA

출력예시1)

1

 

입력예시2)

BAAAAABAA
BAABAAAAAB

출력예시2)

1

 

입력예시3)

A
ABBA

출력예시3)

0

---

코드

 

이 문제는 bfs로 S에서 T를 찾아가는 형식으로 풀면 ,,, 시간초과 나서 못푼다.

거꾸로 T에서 S를 찾아가는 방식으로 해야 풀기 쉽다.

 

문제의 조건이 두 개 밖에 없었기 때문에 dfs재귀함수를 짜면 쉽다.

 

코드 ㄱㄱ

 

import sys; input = sys.stdin.readline # input함수 재정의

def dfs(t, s): # dfs함수 정의
    if t == s: # 만약 t와 s가 같다면
        print(1) # 1출력 후
        exit(0) # 프로그램 종료
    if len(t) == 0: return 0 # 만약 t의 길이가 0이라면 0 반환
    if t[-1] == 'A': dfs(t[:-1], s) # 만약 t의 맨 뒤 원소가 A라면 A제거 후 dfs함수 재귀호출
    if t[0] == 'B': dfs(t[1:][::-1], s) # 만약 t의 맨 앞 원소가 B라면 B제거 후 reverse해서 dfs함수 재귀호출

S = input().rstrip() # 줄바꿈 문자 제거 후 S에 저장
T = input().rstrip() # 줄바꿈 문자 제거 후 T에 저장

dfs(T, S) # dfs함수 호출
print(0) # dfs함수에서 프로그램 종료를 못했다면, S로 T를 만들지 못 한 것 이기 때문에 0 출력

---

실행 화면

 

---

채점 결과