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

문제 링크

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

프로그래머스

 

 

문제 설명

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];
        
        // arr1의 행 수만큼 반복
        for (int i = 0; i < arr1.length; i++) {
            // arr2의 열 수만큼 반복
            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;
    }
}

코드 설명

주석 참고

• 이 문제는 주어진 두 배열을 행렬 곱셈을 사용하여 곱한 결과를 반환하는 문제다.
• 행렬의 곱셈 결과를 저장할 answer를 선언한다. m * k 행렬과 k * n 행렬의 곱은 m * n 크기의 행렬로 반환된다. 따라서 answer의 행은 arr1의 행 크기, 열은 arr2의 열 크기만큼의 사이즈를 가진다.
• 두 행렬의 곱셈을 하기 위해 중첩 for 문을 사용한다.
• 첫 번째 반복문은 arr1 배열의 행 수만큼 반복하고, 두 번째 반복문은 arr2의 열 수만큼 반복한다.
• 마지막으로 행렬의 곱셈을 위해 arr1 배열의 열 수만큼 반복한다.
• 세 번째 반복문에서 각각의 요소들을 곱한 뒤 합한다.
  • answer[i][j] += arr[i][k] * arr2[k][j]
  • answer[0][0] = (arr1[0][0] * arr2[0][0]) + (arr1[0][1] * arr[1][0])