[백준] A와 B(12904) - python

✏️ 문제

수빈이는 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을 출력한다.

🎈 입출력 예

<입력>
B
ABBA

<출력>
1

<입력>
6 10
AB
ABB

<출력>
0

👩‍💻 내 코드

이 문제는 전에 풀었던 백준 A → B와 비슷한 문제이다. 때문에 아이디어를 쉽게 생각해서 풀 수 있었다. 문제에 나와있는 두 가지 연산을 거꾸로 풀면 된다.

문자열 끝에서 부터

• A를 제거한다.
• B를 제거한 후 문자열을 뒤집는다.

위와 같은 아이디어로 아래와 같이 코드를 구현했다.

S = str(input())
T = str(input())

result = 1
while S != T:
    if len(T) == 0:
        result = 0
        break
    else:
        tmp = T[-1]
        if tmp == 'A':
            # A 제거
            T = T[:len(T)-1] 
            continue
        else:
            # B 제거 후 뒤집기
            T = T[:len(T)-1]
            T = T[::-1]

print(result)

처음에는 T를 리스트로 만들었다가 다시 문자열로 바꿔서 S와 비교를 할까 고민했다. 하지만 리스트를 만드는 것 자체가 메모리 소비도 되고 비효율적인 것 같아 문자열 자체를 계속 변경하는 식으로 구현했다. 이전에 푼 문제와 비슷한 문제라 빠르게 접근하여 풀 수 있었다.ㅎㅎ

이 문제를 풀면서 새로 알아낸 점

• 문자열 자체로는 pop()이 불가능하다. (리스트만 가능)
• 문자열 자체로는 reverse()가 불가능하다. (리스트만 가능)
• 문자열을 쉽게 뒤집으려면 [::-1]와 같이 인덱싱하여 뒤집어야 한다.