[프로그래머스] '행렬의 곱셈' 문제 자바스크립트 알고리즘

---

https://school.programmers.co.kr/learn/courses/30/lessons/12949

Q. 행렬의 곱셈

---

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

참고

arr1 row의 각 인덱스와 arr2의 colum의 각 인덱스를 곱해서 더해주면 됨
ex result[0] = (arr1[0][0] arr2[0][0]) + (arr1[0][1] arr2[1][0]) + (arr1[0][2] * arr2[2][0])

• 코드

function solution(arr1, arr2) {
  const answer = [];

  for (let i = 0; i < arr1.length; i++) {
    const result = [];
    for (let k = 0; k < arr2[0].length; k++) {
      let mul = 0;
      for (let j = 0; j < arr1[0].length; j++) {
        mul += arr1[i][j] * arr2[j][k];
      }
      result.push(mul);
    }
    answer.push(result);
  }
  return answer;
}

더 짧은 코드

function solution(arr1, arr2) {
  return arr1.map((row) =>
    arr2[0].map((_, i) => row.reduce((sum, cell, j) => sum + a * arr2[j][i], 0))
  );
}