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

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

프로그래머스

---

 

문제

---

 

풀이

행렬의 곱셈 식만 알면 쉽게 풀 수 있는 문제다!

 

i * k 행렬 A와 k * j 행렬 B를 곱하면 i * j 행렬 AB가 나온다.

AB 행렬 i행 j열 값은 행렬 A의 i행과 행렬 B의 j열을 곱해서 더한 값이 된다.

 

 

소스코드

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