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

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

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

[문제]

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

 

[제한 조건]

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

 

[코드]

class Solution {
    public int[][] solution(int[][] arr1, int[][] arr2) {
        /*
         * 행렬의 곱셉(A행렬 X B행렬)은
         * A 행렬의 열과 B행렬의 행의 크기가 같아야 가능하다.
         */
		
		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;
    }
}

 

[고찰]

 이번 문제는 행렬의 곱셈을 어떻게 하는지만 알고 있다면 쉽게 해결할 수 있는 문제였다. 행렬의 곱셈에서 가장 중요한 것은 두 행렬의 크기인데 A X B 행렬을 수행한다고 가정하면 A 행렬의 열의 크기와 B 행렬의 행의 크기가 같아야 곱셈이 가능하다. 이 점만 주의하면서 3중 for문을 사용하면 된다.