leetcode - reshape the matrix(kotlin)

level - easy

자세한 설명과 내용은 아래 링크를 참고해주세요.
https://velog.io/@mdok1112/leetcode-reshape-the-matrixjava

위의 링크에서는 설명과 자바코드가 들어있으며
여기서는 코틀린으로 작성한 코드만 있습니다.

class Solution {
    fun matrixReshape(mat: Array<IntArray>, r: Int, c: Int): Array<IntArray> {
        if(r*c != mat.size*mat[0].size) {
            return mat
        }

        val result = Array(r){IntArray(c)}
        var i = 0
        var j = 0
        for(matrix in mat) {
            for(m in matrix) {
                if(j == c) {
                    j = 0
                    i++
                }
                result[i][j++] = m
            }
        }

        return result
    }
}