본문 바로가기
코딩테스트(알고리즘)/프로그래머스

[프로그래머스] 행렬의 곱셈 (Javascript)

by Cafe Mocha 2022. 6. 18.

코딩테스트 연습 - 행렬의 곱셈 | 프로그래머스 (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;
}