[Algorithm] 프로그래머스 - 문자 반복 출력하기 (Kotlin)

문자열 my_string과 정수 n이 매개변수로 주어질 때, my_string에 들어있는 각 문자를
n만큼 반복한 문자열을 리턴하는 solution()을 완성하라

 

 

먼저 주먹구구식으로 풀어본 결과 O(n^2)의 시간복잡도를 갖는 코드가 만들어졌다.

 

class Solution {
    fun solution(str: String, n: Int): String {
        val a = str.split("")
        val s = StringBuilder()
        for (i in a) {
            for (j in 0 until n) {
                s.append(i)
            }
        }
        return s.toString()
    }
}

 

이걸 좀 더 간단하게 만들려면 어떻게 할까 생각해봤지만 반복할 때 쓰는 repeat()를 사용하고 문자열을 붙이는 함수를 어떤 걸 어떻게 써야 할지 고민하는 부분에서 막혔다.

정답 제출 후 확인해본 결과 이렇게도 풀 수 있었다.

 

class Solution {
    fun solution(myString: String, n: Int) = myString.map { it.toString().repeat(n) }.joinToString("")
}

 

joinToString()이 무엇인지는 아래 공식문서를 참고한다.

 

https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/join-to-string.html

joinToString - Kotlin Programming Language

구분 기호를 사용하고 제공된 경우 지정된 접두사 및 접미사를 써서 구분된 모든 요소에서 문자열을 만든다. 컬렉션이 거대할 경우 limit의 음수가 아닌 값을 지정할 수 있다. 이 경우 첫 번째 limit 요소만 추가되고 잘린 문자열(기본 "...")이 추가된다

fun <T> Array<out T>.joinToString(
    separator: CharSequence = ", ",
    prefix: CharSequence = "",
    postfix: CharSequence = "",
    limit: Int = -1,
    truncated: CharSequence = "...",
    transform: ((T) -> CharSequence)? = null
): String

 

이것이 joinToString()의 원형이다. 다른 곳에선 어떻게 설명하는지 확인해 봤다.

 

https://notepad96.tistory.com/90

Kotlin joinToString - List 문자열 변환 List to String

리스트를 출력하면 toString()이 호출되며 [...] 같은 형태로 변환된다. 리스트의 원소들을 모두 연결한 하나의 문자열로 만들고 싶을 때 사용할 수 있는 게 joinToString()이다

separator : 원소 간 연결 시 사이에 들어갈 문자열. ""을 넣으면 공백 없이 연결된다
prefix : 만들 문자열에서 시작할 문자열
postfix : 만들 문자열에서 종료할 문자열
limit : 개수 제한