[Python] 12904. A와 B

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

12904번: A와 B

 

접근법

• S → T를 만들 수 있는 것을 구하는 것이 아니라 T→ S가 될 수 있는지를 구한다.
• 위처럼 하는 이유는 T의 마지막 원소를 보면 수행할 수 있는 일이 1개이기 때문이다.

def input_func():
    S = list(input())
    T = list(input())
    return S, T

def edit_string(arr):
    if arr[-1] == 'A':
        arr.pop()
    elif arr[-1] == 'B':
        arr.pop()
        arr.reverse()
    return arr


def solution(S,T):
    while(len(T) > len(S)):
        T = edit_string(T)
    if S == T:
        print("1")
    else:
        print("0")

def main():
    S, T = input_func()
    solution(S,T)

if __name__ == '__main__':
    main()

 

---

사용된 알고리즘

구현, 문자열, 그리디 알고리즘