[프로그래머스 Level.2] 행렬의 곱셈 (연습문제) (Java)

문제 링크

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

프로그래머스

 

 

코딩테스트 연습 > 연습문제 > 행렬의 곱셈

 

문제 설명

 

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

 

제한 조건

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

 

입출력 예

arr1	arr2	return
[[1, 4], [3, 2], [4, 1]]	[[3, 3], [3, 3]]	[[15, 15], [15, 15], [15, 15]]
[[2, 3, 2], [4, 2, 4], [3, 1, 4]]	[[5, 4, 3], [2, 4, 1], [3, 1, 1]]	[[22, 22, 11], [36, 28, 18], [29, 20, 14]]

 

 

 

나의 코드

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

 

 

풀이

1. 행렬의 곱셈을 정확히 알면 풀 수 있는 문제이다.
2. A행렬이 (m*k) , B행렬이 (k*n) 이라면 AB는 (m*n)이 된다.
3. 테스트 케이스 1로 예를 들자면 A 행렬이 [[1, 4], [3, 2], [4, 1]] 이고 B 행렬이 [[3, 3], [3, 3]] 이므로 AB는 (3*2) 행렬이 되고 [[(1*3) + (4*3), (1*3) + (4*3)], [(3*3) + (2*3), (3*3) + (2*3)], [(4*3) + (1*3), (4*3) + (1*3)]]이 된다.