[프로그래머스] 행렬의 곱셈 - 자바스크립트

[ 문제 설명 ]

코딩테스트 연습 - 4주차

---

[ 아이디어 ]

1. 일단 answer 배열에 크기에 맞게 0으로 초기화
2. 2중 for문으로 돌고 각 행렬의 곱셈은 배열 메소드 체이닝을 통해 처리

[ 코드 ]

function solution(arr1, arr2) {
  var answer = Array.from(Array(arr1.length), () => Array(arr2[0].length).fill(0));

  for (let i = 0; i < arr1.length; i++) {
    for (let j = 0; j < arr2[0].length; j++) {
      answer[i][j] = arr1[i].map((item, index) => item * arr2[index][j]).reduce((a, b) => a + b, 0);
    }
  }

  return answer;
}

 

[ 다른 분 코드 ]

1. 극강의 메소드 체이닝을 활용한 예

function solution(arr1, arr2) {
    return arr1.map((row) => arr2[0].map((x,y) => row.reduce((a,b,c) => a + b * arr2[c][y], 0)))
}

 

2. 가장 직관적인 3중 for문 예

function solution(arr1, arr2) {
    var answer = [];
    arr1.forEach(row => answer.push(Array(arr2[0].length).fill(0)));

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

 

 

---