12904번: A와 B

문제 출처 : https://www.acmicpc.net/problem/12904

12904번: A와 B

언어 : Kotlin

 

문제 설명 :

수빈이는 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을 출력한다.

 

제한 사항 :

• 시간 제한 : 2초
• 메모리 제한 : 512MB

 

입출력 예 :

입력	출력
B ABBA	1
AB ABB	0

 

풀이 : 

처음엔 S를 T로 만드는 방법을 생각했는데, T를 주어진 연산을 반대로 진행하는 방법이 떠올랐다.

import java.io.BufferedWriter
import java.io.OutputStreamWriter

fun main() = with(System.`in`.bufferedReader()) {
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val s = readLine()
    var t = readLine()
    while (true) {
        if (s.length == t.length || t.length == 1) break

        val cur = t.length - 1
        if (t[cur] == 'A') t = t.substring(0, cur)
        else if (t[cur] == 'B') t = t.substring(0, cur).reversed()
    }

    bw.write("${if (s == t) 1 else 0}")
    bw.flush()
    bw.close()
}