[Algorithm] 프로그래머스 - 그림 확대 (Kotlin)

직사각형 형태의 그림 파일이 있고 이 파일은 1x1 크기의 정사각형 픽셀로 이뤄져 있다. 이 파일을 나타낸 문자열 배열 picture, 정수 k가 매개변수로 주어질 때, 이 파일을 가로와 세로로 k배 늘린 파일을 나타내도록 문자열 배열을 리턴하는 함수를 완성하라

 

 

풀이법이 떠오르지 않아서 챗지피티에 확인해 봤다.

 

class Solution {
    fun solution(picture: Array<String>, k: Int): Array<String> {
        val result = mutableListOf<String>()

        for (row in picture) {
            val expandedRow = row.flatMap { char -> List(k) { char } }.joinToString("")
            repeat(k) {
                result.add(expandedRow)
            }
        }

        return result.toTypedArray()
    }
}

 

flatMap은 리스트 안의 요소들을 다른 리스트로 매핑한 다음 모든 리스트를 하나로 평평하게 만드는 함수다.

List(k) { char }는 k개의 char를 가진 리스트를 생성한다.

때문에 expandedRow에는 행의 각 문자를 k번 반복해서 새 리스트를 생성한 후 하나로 만든다.

아래는 예시 코드다.

 

fun main() {
    val row = "x.x"
    val k = 3

    // 각 문자를 k번 반복하는 리스트로 변환한 후 평평하게 만든 리스트
    val expandedRow = row.flatMap { char -> List(k) { char } }.joinToString("")
    println(expandedRow) // "xxx...xxx"
}

 

char가 x고 k가 3이면 List(3) { char }는 ['x', 'x', 'x']를 생성한다.

점과 나머지 x에도 동일한 처리가 가해지면 "xxx...xxx"를 출력하는 것이다.

마지막에 joinToString("")을 써서 중간 공백이 없는 문자열로 만드는데, 이 작업을 solution()의 2번째 매개변수인 k번만큼 반복한다.

모든 반복이 끝나면 result를 Array로 캐스팅해서 리턴하면 해결된다.