12904: A와 B - Kotlin

12904번: A와 B

문제

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

 

나의 풀이 - 1차 실패(시간초과)

처음 생각한 방식은 dfs를 사용해서 두 가지 연산을 처리하는 방식이었다. 조금 불안했지만 역시나 시간 초과가 발생하였다. 그래서 이 문제는 그리디하게 풀어야 겠다는 생각이 들었고 고민을 하였다.

import java.util.*
import kotlin.system.exitProcess

fun main() = with(System.`in`.bufferedReader()) {
    var s = readLine()
    val t = readLine()

    dfs(s, t)
    println(0)
}

fun dfs(s: String, t: String) {
    if (s == t) {
        println(1)
        exitProcess(0)
    }

    if (s.length > t.length) return

    dfs(s + 'A', t)
    dfs(s.reversed() + 'B', t)
}

 

 

나의 풀이 - 2차 성공

그리디 하게 풀 수 있는 방식이 뭐가 있을까 고민하다가 찾은 방법은 s에서 t로 가는 것을 고려하는 것이 아니라 t에서 s로 가는 것을 고려하는 것이다. s에서 할 수 있는 연산은 어느 상황에던 2가지이다. 반면 t는 마지막 문자가 'A' 라면 s에서 이전에 'A'를 추가했다는 말이고 마지막 문자가 'B'라면 s에서 이전에 'B'를 추가했다는 말이 된다. 그렇기 때문에 t가 s의 길이가 될 때까지만 반복하면 s가 t가 될 수 있는지 판단할 수 있다.

fun main() = with(System.`in`.bufferedReader()) {
    val s = readLine()
    var t = readLine()

    while (t.length >= s.length) {
        if(t == s){
            println(1)
            return@with
        }

        if (t.last() == 'A')
            t = t.substring(0 until t.lastIndex)
        else
            t = t.substring(0 until t.lastIndex).reversed()
    }

    println(0)
}