[백준 12904번] A와 B - 파이썬(python)

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를 T로 바꾸는 경우보다 T를 S로 바꾸는 것이 휠씬 편하다.
• 변경된 S와 T가 동일하다면 1을 출력하고 두 개의 길이가 같거나 작을 때까지 동일하지 않다면 0을 출력한다.
• 마지막이 A이면 A를 제거한 나머지를 저장한다.
• 마지막이 B하면 B를 제거하고 뒤집는다.

코드

S = input()
T = input()

while True:
    # T와 S가 동일한 경우 1을 출력하고 break
    if T == S:
        print(1)
        break
    # T와 S가 동일하지 않은데 길이가 같거나 적다면 0을 출력하고 break
    if len(T) <= len(S):
        print(0)
        break
    # 마지막이 A라면 A떼기
    if T[-1] == 'A':
        T = T[:-1]
    # B라면 B를 떼고 뒤집기
    elif T[-1] == 'B':
        T = T[:-1][::-1]