[프로그래머스] 행렬의 곱셈 - JAVA

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12949

프로그래머스

 

풀이

이그림을 잘 기억하자

 

예를 들어 2 x 3 행렬 A와 3 x 2 행렬 B가 있을 때

두 행렬의 곱셈 C는 다음과 같이 나타낼 수 있다

 

코드

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