백준12904 문제 보러가기

문제

수빈이는 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 문자열에서 뒤에서부터 문자를 하나씩 빼서 S를 만드는 과정으로 해결을 시도했다.

T를 list로 담고, pop을 하면 맨 뒤 index의 문자부터 빼내기 때문에 그 때 'A'라면 그대로, 'B'라면 문자열을 뒤집으면 된다.

그래서 사실 코드는 간단하다.
그런데 여기서 사실 나의 실수 때문에 시간을 오래 잡아 먹었다.

S = input()
T = list(input())

while(1) :
    if len(T) == len(S) :
        if T == S : # 이 부분이 틀림
            print(1)
            break
        else :
            print(0)
            break
        
    check = T.pop()

    if check == 'A' :
        T = T
    else :
        T.reverse()

여기서는 T가 list인데 계속 S와 비교해서 틀린 답이 나왔다.
T에서 다시 문자열을 만들어야 했다.

최종 코드

S = input()
T = list(input())
ch = ''
while(1) :
    if len(T) == len(S) :
        for i in T :
            ch += i
            
        if ch == S :
            print(1)
            break
        else :
            print(0)
            break
        
    check = T.pop()

    if check == 'A' :
        T = T
    else :
        T.reverse()