백준 온라인 저지, 그리디 / 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을 출력한다.

제한

예제 입력 1

B
ABBA

예제 출력 1

1

예제 입력 2

AB
ABB

예제 출력 2

0

힌트

접근 방법

- S가 T가 될 수 있는지 확인하기 위해 T에서 현재 문자열의 형태를 보고 제일 뒤에 B가 있다면 이를 지우고 T를 뒤집어준다. 혹은 제일 뒤에 A가 있다면 이를 지워준다.
- S와 T가 같은 길이가 되었을 때 위의 동작을 멈추고 만약 둘이 같다면 1, 다르다면 0을 출력한다.
- 시간 복잡도: O(N**2)

코드

# https://www.acmicpc.net/problem/12904
# 접근 방법
# S가 T가 될 수 있는지 확인하기 위해 T에서 현재 문자열의 형태를 보고 제일 뒤에 B가 있다면 이를 지우고 T를 뒤집어준다. 혹은 제일 뒤에 A가 있다면 이를 지워준다.
# S와 T가 같은 길이가 되었을 때 위의 동작을 멈추고 만약 둘이 같다면 1, 다르다면 0을 출력한다.
# 시간 복잡도: O(N**2)
s = [x for x in input()]
t = [x for x in input()]
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)