Algorithm

[JS] 프로그래머스 lv2. 행렬의 곱셈

judahhh 2023. 1. 29. 21:15
728x90

문제 설명

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) {
   let answer = [],sum=0
   for(let i=0;i<arr1.length;i++){
    let arr=[]
     for(let j=0;j< arr2[0].length;j++){
        sum=0
      for(let k = 0; k < arr1[0].length; k++){
           sum+=arr1[i][k]*arr2[k][j]   
        }
         arr.push(sum)
      }
      answer.push(arr)
    }
    return answer
}

풀이 전략

행렬을 직접 써서 반복문을 인덱스로 직접 나타내보는게 훨씬 직관적인 문제였다.

꼭꼭 행렬을 직접 그려서 구현해보기!!!

728x90