문제70. 행렬 곱하기

 

 

행렬 2개가 주어졌을때,

• 곱할 수 있는 행렬인지 확인하고 곱할수 있다면 그 결과를 출력하고, 곱할 수 없다면 -1을 출력하는 프로그램을 만들어보자

 

예시

// 입력
a = [
  [1, 2],
  [3, 4]
];
b = [
  [4, 5],
  [6, 7],
];

// 출력
[
  [16, 19],
  [36, 43]
];

 

 

 

미리 생각해 봐야할 것

행렬의 곱셈법

(a의 1행1열(1) * b의 1행1열(4)) + (a의 1행2열(2) * b의 2행1열(6)) = 16
(a의 1행1열(1) * b의 1행2열(5)) + (a의 1행2열(2) * b의 2행2열(7)) = 19
(a의 2행1열(3) * b의 1행1열(4)) + (a의 2행2열(4) * b의 2행1열(6)) = 36
(a의 2행1열(3) * b의 1행2열(5)) + (a의 2행2열(4) * b의 2행2열(7)) = 43

 

 

 

 

답

let as = [
  [1, 2],
  [3, 4]
];
let bs = [
  [4, 5],
  [6, 7],
];

function solution(A, B) {
    let answer = [];
    for (let i = 0; i < A.length; i++) {
      // 행렬 A의 행 접근 - A
      const row = A[i];
      answer.push([]);
      // 행렬 B의 열 접근해서 곱하기 - B
      // 행렬 B의 열 길이
      for (let j = 0; j < B[0].length; j++) {
        let sum = 0;
        // 행렬 B의 행 길이
        // B는 열을 고정해놓고 행을 이동하기 - C
        for (let k = 0; k < B.length; k++) {
          sum += row[k] * B[k][j];
        }
        answer[i].push(sum);
      }
    }
    return answer;
  }

console.log(solution(as, bs))