BOJ [A와 B 2]

문제

2022-05-05

문제 보기

두 문자열 S와 T가 주어졌을 때, S를 T로 바꾸는 게임이다.

문자열을 바꿀 때는 다음과 같은 두 가지 연산만 가능하다.

문자열의 뒤에 A를 추가한다.

문자열의 뒤에 B를 추가하고 문자열을 뒤집는다.

주어진 조건을 이용해서 S를 T로 만들 수 있는지 없는지 알아내는 프로그램을 작성하시오.

풀이

짧은 문자열 S 로부터 경우의 수를 나누어 T를 만들려 하면 최대 나올 수 있는 경우가

2^50 이므로 겹치는 경우를 제외하더라도 너무 많은 경우가 나온다.

하지만 긴 문자열 T로부터 S를 만들면 문자열에 따라

BKA => (AK, BK)

AKA => (AK)

BKB => (BK)

AKB => ( )

다음과 같이 경우의 수가 나뉘므로 탐색하는 가지수가 확 줄어든다.

코드

t = tuple(input())
s = tuple(input())

n = len(s) - len(t)

temp_set = set([])
temp_set.add(s)

for _ in range(n):

	temp = set([])
	
	for i in temp_set:
		
		i = list(i)
		
		if i[-1] == 'A':
			del i[-1]
			temp.add(tuple(i))
			i.append('A')

		if i[0] == 'B':
			i.reverse()
			del i[-1]
			temp.add(tuple(i))
	
	temp_set = set(temp)

if t in temp_set:
	print(1)
else:
	print(0)