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

 

문제

프로그래머스

 

알고리즘

없음

 

풀이

a x b 크기의 행렬과 b x c 크기의 행렬의 곱은 a x c 크기의 행렬로 표현될 수 있고,

c[i][j] = arr1[i][0] * arr2[0][j] + arr1[i][1] * arr2[1][j] + ... + arr1[i][b] * arr2[b][j]  이다.

 

출처: http://db.kockoc.com/lec/Hard/g1.htm

 

코드

class Solution {
    public int[][] solution(int[][] arr1, int[][] arr2) {
        int[][] answer = new int[arr1.length][arr2[0].length];
        
        for(int i=0; i<arr1.length; i++) {
            for(int j=0; j<arr2[0].length; j++) {
                int sum = 0;
                
                for(int k=0; k<arr1[0].length; k++)
                    sum += arr1[i][k] * arr2[k][j];
                
                answer[i][j] = sum;
            }
        }
        return answer;
    }
}