[Algorithm] 프로그래머스 - qr code (Kotlin)

정수 q, r과 문자열 code가 주어질 때 code의 각 인덱스를 q로 나눴을 때 나머지가 r인 위치의 문자를 앞에서부터 순서대로 이어 붙인 문자열을 리턴하는 함수를 완성하라

 

 

forEachIndexed를 쓰면 쉽게 풀 수 있다. 나머지가 r인 경우에만 StringBuilder.append()로 이어붙이고 forEachIndexed가 끝날 때 StringBuilder를 리턴해주면 된다.

 

class Solution {
    fun solution(q: Int, r: Int, code: String): String {
        val sb = StringBuilder()
        code.forEachIndexed { index, c ->
            if (index % q == r) {
                sb.append(c)
            }
        }

        return sb.toString()
    }
}