👩‍💻 코테 공부/코테 공부

[코테 - py] 12904번 A와 B

수댕ʕت̫͡ʔ 2024. 6. 1. 22:42

1. 문제

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

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

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

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

2. 입력

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

3. 출력

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

4. 예제 입력1

B
ABBA

5. 예제 출력1

1

6. 내가 푼 답안

data1 = list(input())
data2 = list(input())

while len(data1) != len(data2):
    # 만약에 맨 뒤가 A라면 제거
    if data2[-1] == 'A':
        data2.pop()
    # 만약에 맨 뒤가 B라면 제거 후 뒤집기
    elif data2[-1] == 'B':
        data2.pop()
        data2.reverse()
    # 만약에 data1과 data2가 같다면 종료
    if data1 == data2:
        print(1)
        exit()

# 같지 않다면 0 출력
print(0)