[Lv. 2] 행렬의 곱셈

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/12949

프로그래머스

난이도 : Level.2

언어 : Kotlin

 

문제 설명 :

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]]

풀이 : 

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

        repeat(arr1.size) { i ->
            repeat(arr2[0].size) { j ->
                repeat(arr2.size) { k ->
                    answer[i][j] += arr1[i][k] * arr2[k][j]
                }
            }
        }

        return answer
    }
}

 

제한 조건에서 곱할 수 있는 배열만 주어진다 했으니, 그냥 하면 될 듯하다.

곱할 수 없을 경우 [-1]을 반환하라고 해도 try/catch로 잡아서 리턴하면 되지 않을까 싶기도 하고