행렬의 곱셈

문제 설명

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

문제 풀이

function solution(arr1, arr2) {
  const n = arr1.length;
  const m = arr2.length;
  const k = arr2[0].length;

  const result = Array.from({ length: n }, () => Array(k).fill(0));

  for (let i = 0; i < n; i++) {
    for (let j = 0; j < k; j++) {
        for (let t = 0; t < m; t++) {
            result[i][j] += arr1[i][t] * arr2[t][j]
        }
    }
  }

  return result
}

아이디어

1. arr1, arr2의 길이와 arr2[0]의 길이 저장
2. 반복문 돌려가면서 arr1과 arr2의 곱을 누적합