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

🎈문제

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

 

🎁어떤 알고리즘?

1. s를 t로 만드려면 A를 뒤에 붙이거나,  뒤집어서 B를 뒤에 붙이는 2가지 경우의 수를 t-s개수 만큼 돌려봐야한다. 하지만 t를 s로 만들 경우 훨씬 간단해진다.

2. t의 맨 뒤 알파벳에 맞춰서 제거하다보면 s와 같아지던지, t의 모든 문자열이 pop 되었음에도 s와 같지 않는 두 가지 상황이 나오게된다. 

 

💻코드

s = list(input())
t = list(input())
same = False

while t:
    if s == t:
        same = True
        break
    if t[-1] == 'A':
        t.pop()
    elif t[-1] == 'B':
        t.pop()
        t = t[::-1]

if same:
    print(1)
else:
    print(0)

 

🎄코드 설명

1. 문자열을 수정해야 할 때는 항상 list형이여야 함을 기억하자.

2. 따라서 s와 t값을 list로 받는다.

3. t의 맨 뒤 문자열을 하나씩 뽑아가며 s와 같아지는지를 확인할 것 이므로 while의 조건은 t가 되고, t가 모두 뽑혀나가 빈 리스트가 되면 while문은 종료.

4. 리스트 t의 맨 뒤 값이 'A'일 경우 단순히 뽑아내고, 'B'일 경우에는 뽑아내고 뒤집는다.

5. s 와 t가 같을 때 탈출하는 탈출 조건을 넣어준다.