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

programmers.co.kr/learn/courses/30/lessons/12949

코딩테스트 연습 - 행렬의 곱셈

문제풀이

행렬의 곱은

arr 1             arr2        = answer

[a1,b1]                           [a1*c1 + b1*c2,  a1*d1+b1*d2]

[a2,b2]    x    [c1,d1]     =  [a2*c1 + b2*c2,  a2*d1+b2*d2]

[a3,b3]         [c2,d2]         [a3*c1 + b3*c2,  a3*d1+b3*d2]

로 할 수 있습니다. 정리하자면 answer의 행렬크기는 3x2 2x2 arr1의 행과 arr2의 열로 정의할 수 있습니다.

arr1의 행과 열은 기존의 2차원 배열 검색하듯이 하면서 arr2의 열만 반복문으로 하면된다. arr1의 열이 증가할때 같이 arr2의 행을 증가시키면서 더하면 해결된다.

코드

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 k=0; k<arr2[0].length; k++){
                for(int j =0; j<arr1[0].length; j++){
                    answer[i][k] += arr1[i][j] * arr2[j][k];
                }
            }
        }
        
        return answer;
    }
}