🗃️javascript/프로그래머스

행렬의 곱셈(js.Lv.2)

하얀성 2024. 6. 27. 09:23

2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 함수, solution을 완성해주세요.

제한 조건
  • 행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.
  • 행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.
  • 곱할 수 있는 배열만 주어집니다.
입출력 예arr1arr2return
[[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[0].length로 표현된 같은 공간의 값만큼 k번 반복하는 것.

function solution(arr1, arr2) {
    let n = arr1.length
    let n2 = arr2[0].length
    let outArr = new Array(n);  

    for(let i=0;i<n; i++){//행 길이만큼 반복
        let inArr = new Array(n2);
        for(let j=0; j<n2; j++){ //두번째 내부 길이만큼 반복
           let sum = 0;
            for(let k =0; k<arr1[0].length ;k++){ //첫번째 내부 길이만큼 반복
                sum += arr1[i][k] * arr2[k][j]

            }
            inArr[j] = sum;
        }
        outArr[i] = inArr
    }

    return outArr
}