[JS] 연습문제 - 행렬의 곱셈

function solution(arr1, arr2) {
  // arr1 -> p * q , arr2 -> q * r
  let p = arr1.length;
  let q = arr2.length;
  let r = arr2[0].length;

  let answer = new Array(p);
  for (let i = 0; i < p; i++) {
    answer[i] = new Array(r).fill(0);
  }

  for (let i = 0; i < p; i++) {
    let row1 = arr1[i];
    for (let j = 0; j < q; j++) {
      let first = row1[j];
      let row2 = arr2[j];
      for (let k = 0; k < r; k++) {
        let second = row2[k];
        answer[i][k] += first * second;
      }
    }
  }
  return answer;
}

 

프로그래머스

 

1. 주어진 두 행렬의 행과 열에 대한 길이를 각각 p, q, r로 선언 및 할당합니다.(이 때, q는 첫번째 행렬의 열이면서 두번째 행렬의 행)
2. p * q 행렬과 q * r 행렬의 곱은 p * r 행렬이므로 그에 맞게 answer 배열을 선언 및 할당해줍니다. 이 때, 각 배열의 값에는 0을 채워줍니다.(추후 반복문을 통해 곱한 값을 더할 때, 반복적으로 덧셈을 하기 위함)
3. 처음 p를 통해 첫번째 행렬의 i행을 선언 및 할당, q를 통해 i행의 j열 즉, 첫번째 행렬의 요소를 first로 선언합니다. 이 때, 동시에 두번째 행렬의 j행을 선언 및 할당합니다.
4. 다음 r을 통해 j행의 k열 즉, 두번째 행렬의 요소를 second로 선언합니다. 또한, 앞선 first(첫번째 행렬의 값)과 second(두번째 행렬의 값)을 곱한 것을 answer[i][k]에 더하기를 반복합니다.

 

행렬의 곱셈을 표현하기 위해 삼중반복문을 쓰다보니까 다소 헷갈리는데, i j k 에 대해 천천히 손으로 써가면서 눈으로 보면 어떻게 이게 행렬의 곱셈이 되는지 알 수 있다.
알고리즘 문제들 중에 '와 이건 컴퓨터가 계산하는 게 훨씬 편하네' 싶은 게 많은데, 이 문제는 '와 이건 내가 계산하는 게 더 편하겠다'싶은 문제였다.ㅋㅋ