BOJ / python3 백준 12904번 : A와 B

https://www.acmicpc.net/problem/12904

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

 

SOLUTION

 

1. 문자열 s를 t로 바꾸는 문제로, 두가지 조건이 주어진다.

첫째, 문자열의 뒤에 A를 추가

둘째, 문자열을 뒤집고 뒤에 B를 추가 

두가지 방법을 이용하여 s를 t로 바꿀 수 있는지 없는지 여부를 판단한다. 

 

2. s에서 t로 문자를 하나씩 추가하는 것보다 t에서 하나씩 제거하면서 s가 될 수 있는지를 판단해보자. 

문자열의 마지막 문자를 기준으로 

 

마지막 문자가 A라면 그냥 A를 추가한 것이므로 A제거.

마지막 문자가 B라면 문자열을 뒤집은 뒤 B를 추가한 것이므로 B를 제거한 뒤 문자열을 뒤집어준다. 

 

위 과정을 문자열 t의 길이가 s와 같아질 때까지 반복하여

s와 t가 같은 문자라면 1을 출력

다르다면 0을 출력해준다. 

 

소스코드

import sys
s = sys.stdin.readline().rstrip()
t = sys.stdin.readline().rstrip()

while len(t) > len(s):
    if t[-1] == 'A':
        t = t[:-1]
    else:
        t = t[:-1]
        t = t[::-1]

if s == t:
    print(1)
else:
    print(0)