프로그래머스 lv0 잘라서 배열로 저장하기 (Kotlin)

https://school.programmers.co.kr/learn/courses/30/lessons/120913

프로그래머스

문제 설명

문자열 my_str과 n이 매개변수로 주어질 때, my_str을 길이 n씩 잘라서 저장한 배열을 return하도록 solution 함수를 완성해주세요.

제한사항

• 1 ≤ my_str의 길이 ≤ 100
• 1 ≤ n ≤ my_str의 길이
• my_str은 알파벳 소문자, 대문자, 숫자로 이루어져 있습니다.

입출력 예

my_str	n	result
"abc1Addfggg4556b"	6	["abc1Ad", "dfggg4", "556b"]
"abcdef123"	3	["abc", "def", "123"]

 풀이

class Solution {
    fun solution(my_str: String, n: Int): MutableList<String> {
        val answer = mutableListOf<String>()

        for (i in my_str.indices step (n)) {
            if (my_str.length >= i + n) {
                answer.add(my_str.substring(i, i + n))
            } else {
                answer.add(my_str.substring(i))
            }
        }
        return answer
    }
}

다른풀이

class solution {
	fun solution(my_str: String, n: Int) = my_str.chunked(n).toTypedArray()
}

chunked : 개수를 기준으로 문자열을 자르는 함수. 리턴값도 List이다.