[Algorithm] 프로그래머스 - 문자열 바꿔서 찾기 (Kotlin)

문자 "A", "B"로 이뤄진 문자열 myString, pat이 주어진다. myString의 "A"를 "B"로, "B"를 "A"로 바꾼 문자열의 연속하는 부분 문자열 중 pat이 있으면 1, 아니면 0을 리턴하는 함수를 완성하라

 

 

myString에 map 걸어서 글자를 바꾼 다음 공백 없는 문자열로 만들고, 이 문자열에 pat이 있냐 없냐에 따라 1 or 0을 리턴하게 하면 된다.

 

class Solution {
    fun solution(myString: String, pat: String): Int {
        val str = myString.map { 
            when (it) {
                'A' -> 'B'
                'B' -> 'A'
                else -> it
            }
        }.joinToString("")

        return if (str.contains(pat)) 1 else 0
    }
}