[알고리즘] BOJ 12904 A와 B

[BOJ] 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(짧은 문자열)로 변환하는 방식으로 문제 해결 방법을 찾았습니다.
• 문자열 T의 마지막 문자가 'A'일 경우 "문자열의 뒤에 A를 추가한다."가 마지막 연산으로 실행됐을 경우의 수 밖에 없으므로 T 문자열에서 제일 마지막 문자 'A'를 제거하고,
• 만약 문자열 T의 마지막 문자가 'B'일 경우 "문자열을 뒤집고 뒤에 B를 추가한다."가 마지막 연산으로 실행됐을 경우의 수 밖에 없으므로 T문자열에서 제일 마지막 문자 'B'를 제거하고 남은 문자열을 reverse() 해주는 방식으로 진행합니다.
• 문자열 T와 문자열 S의 길이가 같아졌을 때 서로 같은 문자열을 갖고 있다면 '1'을 출력하고 아니라면 '0'을 출력하는 방식으로 코드를 작성하였습니다.

✍ 코드

from sys import stdin

S = list(stdin.readline().rstrip())
T = list(stdin.readline().rstrip())

num = len(T) - len(S) # 문자열 길이 차이

for _ in range(num):
  if T[-1] == 'A': # 맨 뒤에 값이 'A'일 경우
    T.pop()
  else: # 맨 뒤에 값이 'B'일 경우
    T.pop()
    T.reverse()

if "".join(T) == "".join(S): # 연산이 끝난 두 문자열이 같을 경우
  print(1)
else: # 연산이 끝난 두 문자열이 다를 경우
  print(0)