문제
수빈이는 A와 B로만 이루어진 영어 단어 존재한다는 사실에 놀랐다. 대표적인 예로 AB (Abdominal의 약자), BAA (양의 울음 소리), AA (용암의 종류), ABBA (스웨덴 팝 그룹)이 있다.
이런 사실에 놀란 수빈이는 간단한 게임을 만들기로 했다. 두 문자열 S와 T가 주어졌을 때, S를 T로 바꾸는 게임이다. 문자열을 바꿀 때는 다음과 같은 두 가지 연산만 가능하다.
- 문자열의 뒤에 A를 추가한다.
- 문자열의 뒤에 B를 추가하고 문자열을 뒤집는다.
주어진 조건을 이용해서 S를 T로 만들 수 있는지 없는지 알아내는 프로그램을 작성하시오.
풀이
문제의 핵심 : T를 S로 만들자 !
S로 T를 만들기에는 경우의 수가 너무 많고 메모리 차지를 너무 많이 하기 때문에 제거하면서 같은지 확인하는게 편하다.
그렇다면 문제를 다음으로 변경 가능
- 문자열의 뒤에 A를 추가한다. --> 문자열의 뒤에 A가 있으면 제거한다.
- 문자열의 뒤에 B를 추가하고 문자열을 뒤집는다. -> 문자열의 앞에 B가 B를 제거하고 뒤집는다.
+) 1번과 2번을 다 검사해봐야 하기 때문에 3, 4 번째 조건은 if - elif 를 쓰지 말고, if - if를 써야함
- 예시 : s = A / t = BABA
CODE
import sys
s = sys.stdin.readline().strip()
t = sys.stdin.readline().strip()
result = 0
def dfs(t_str):
global result
# 종료 조건
if t_str == s:
result = 1
return
if len(t_str) == 0:
return
if t_str[-1] == 'A':
dfs(t_str[:-1])
if t_str[0] == 'B':
dfs(t_str[1:][::-1])
dfs(t)
print(result)
'CS > Coding Test' 카테고리의 다른 글
[백준] 20437: 문자열 게임 python 구현 (0) | 2023.03.31 |
---|---|
[백준] 20055: 컨베이어 벨트 위의 로봇 python 구현 (0) | 2023.03.30 |
[백준] 13549: 숨바꼭질3 python 구현 / BFS (0) | 2023.03.30 |
[백준] 14503: 로봇 청소기 python 구현 / DFS (0) | 2023.03.23 |
[백준] 9205: 맥주 마시면서 걸어가기 python 구현 / DFS (0) | 2023.03.22 |
댓글