[Kotlin] 행렬의 곱

https://ko.wikipedia.org/wiki/%ED%96%89%EB%A0%AC_%EA%B3%B1%EC%85%88

행렬 곱셈 - 위키백과, 우리 모두의 백과사전

 

행렬 A는 a x b 행렬, 행렬 B는 b x c행렬이라고 할 때, 

행렬 A와 행렬 B의 곱으로 나온 행렬 C의 원소는 다음과 같다.

행렬의 곱셈 후 행렬 요소의 식

 

 

fun matrixMultiply(arr1: Array<IntArray>, arr2: Array<IntArray>): Array<IntArray> {
    val (m1, m2) = if (arr1[0].size == arr2.size) {
        arr1 to arr2
    } else { // arr2[0].size == arr1.size
        arr2 to arr1
    }

    return Array<IntArray>(m1.size) { i ->
        IntArray(m2[0].size) { j ->
            var sum = 0
            for (k in m2.indices) {
                sum += m1[i][k] * m2[k][j]
            }
            sum
        }
    }
}