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

📚 문제 설명

2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 함수, solution을 완성해주세요.

제한사항

• 행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.
• 행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.
• 곱할 수 있는 배열만 주어집니다.

🔎 입출력 예

💻 코드

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

📖 Solution

이 문제는 행렬의 곱셈을 구하는 방법만 알면 쉽게 풀 수 있는 문제이다.
나는 행렬의 곱을 구하는 방법이 헷갈려서 먼저 이론을 보고 풀었다.

행렬의 곱을 구하는 방법은 크게 어렵지 않다.

앞쪽의 열과 뒤쪽의 행이 같아야만 곱셈이 가능하다!
연산 결과 행렬의 크기는 앞쪽의 행 * 뒤쪽의 열 이다.
answer[i][j] += arr1[i][k] * arr2[k][j] 이다.
-> 이 때 k는 앞쪽의 열이자 뒤쪽의 행

위에만 잘 기억해서 코드를 짜면된다 :)

📌 참고자료 - 행렬의 곱셈