본문 바로가기
CS/알고리즘

[연습문제] PGS lv2. 행렬의 곱셈

by westpearl 2025. 9. 14.
728x90
반응형
SMALL

function solution(arr1, arr2) {
    const result = [];
    for(let i=0;i<arr1.length;i++){  
        const arr = [];
        for(let k =0;k<arr2[0].length;k++){ 
            let sum=0;
            for(let j=0;j<arr1[0].length;j++){
                sum += arr1[i][j] * arr2[j][k];   
            }
            arr.push(sum);
        }
    
        result.push(arr)
    }
    return result;
}

 

 

더 간단한 코드 - gpt

function solution(arr1, arr2) {
  return arr1.map(row =>
    arr2[0].map((_, colIndex) =>
      row.reduce((sum, value, i) => sum + value * arr2[i][colIndex], 0)
    )
  );
}

 

LIST