코딩테스트(알고리즘)/프로그래머스
[프로그래머스] 행렬의 곱셈 (Javascript)
Cafe Mocha
2022. 6. 18. 15:00
코딩테스트 연습 - 행렬의 곱셈 | 프로그래머스 (programmers.co.kr)
코딩테스트 연습 - 행렬의 곱셈
[[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]]
programmers.co.kr
Javascript
function solution(arr1, arr2) {
let answer = [];
let ans =[];
let len1 = arr1.length;
let len2 = arr2.length;
let eleLen2 = arr2[0].length;
let psum = 0;
let cnt = 0;
for(let i=0;i<len1;i++){
for(let a=0;a<eleLen2;a++){
for(let j=0;j<len2;j++){
psum+=arr1[i][j]*arr2[j][a];
cnt++
if(cnt===len2){
cnt=0;
ans.push(psum);
psum=0;
if(ans.length===eleLen2){
answer.push(ans);
ans =[];
}
}
}
}
}
return answer;
}