본문 바로가기

TLI/코드카타

2024.06.20 TIL 코트카타 88번(행열의 곱셈)

    fun solution(arr1: Array<IntArray>, arr2: Array<IntArray>): Array<IntArray> {
        var answer = Array(arr1.size, {IntArray(arr2[0].size)})

        for(arrI in arr1.indices) {
            for(arrJ in arr2[0].indices) {
                var sumElements = 0
                for(arrK in arr2.indices) {
                    sumElements +=  arr1[arrI][arrK] * arr2[arrK][arrJ]
                }
                answer[arrI][arrJ] = sumElements
            }
        }
        return answer
    }

 

포인트

- 행렬의 곱셈 공식에 맞춰 반복문을 돌려주면 된다.