04/09 알고리즘 공부(88) - 행렬의 곱셈

조건

2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 함수, solution을 완성해주세요.

제한조건

• 행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.
• 행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.
• 곱할 수 있는 배열만 주어집니다.

입출력 예

arr1	arr2	return
[[1, 4], [3, 2], [4, 1]]	[[3, 3], [3, 3]]	[[15, 15], [15, 15], [15, 15]]
[[2, 3, 2], [4, 2, 4], [3, 1, 4]]	[[5, 4, 3], [2, 4, 1], [3, 1, 1]]	[[22, 22, 11], [36, 28, 18], [29, 20, 14]]

알고리즘 예상

행렬 곱셈

1. 행렬 곱셈은 위 이미지 처럼 첫번째 행렬의 n번째 행과 두번째 행렬의 m번째 열의 원소를 각각 더해 합한 값을 만들어지는 행렬의 n행 m열의 위치에 배치한다.
2. 반복문을 이용해 각 행렬의 곱을 반환해봐야겠다

 

초기 코드

class Solution {
    fun solution(arr1: Array<IntArray>, arr2: Array<IntArray>): Array<IntArray> {
        var answer = arrayOf<IntArray>()
        return answer
    }
}

 

내 코드

class Solution {
    fun solution(arr1: Array<IntArray>, arr2: Array<IntArray>): Array<IntArray> {
        val answer = Array(arr1.size) {//행 갯수
            Array(arr2[0].size) { 0 }.toIntArray() // 열 갯수
        }
        for (i in answer.indices) {
            for (j in answer[0].indices) {
                for (k in arr1[0].indices){
                    answer[i][j] += arr1[i][k] * arr2[k][j]
                }
            }
        }

        return answer
    }
}

 

다른 사람의 풀이

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

        var sum : Int = 0

        for(i in 0..arr1.size-1 step 1){
            for(j in 0..arr2[0].size-1 step 1){

                for(k in 0..arr1[0].size-1)
                {
                    sum += arr1[i][k] * arr2[k][j]
                }
                answer2[i][j] = sum
                sum = 0
            }
        }
        answer = answer2

        return answer
    }
}

 

개선점 또는 배운점

1. IntArray(){}를 두고 Array<Int>(){}.toIntArray()를 내가 왜 썼는지 모르겠다. 
2. sum이라는 변수를 이용해 원소 호출 횟수를 줄였다.

 

개선된 코드

class Solution {
    fun solution(arr1: Array<IntArray>, arr2: Array<IntArray>): Array<IntArray> {
        val answer = Array(arr1.size) {//행 갯수
            IntArray(arr2[0].size) { 0 } // 열 갯수
        }
        for (i in answer.indices) {
            for (j in answer[0].indices) {
                var sum = 0
                for (k in arr1[0].indices) {
                    sum += arr1[i][k] * arr2[k][j]
                }
                answer[i][j] = sum
            }
        }

        return answer
    }
}