[프로그래머스] 행렬의 곱셈 - 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[][] result  = 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++){
                    result[i][j] += arr1[i][k] * arr2[k][j];
                }
            }
        }
        return result;
    }
}

• 행렬의 곱셈을 하는 방법을 알면 쉽게 풀 수 있는 문제이다.
• 변수 int i를 사용하는 제일 바깥쪽 for문은 arr1에 행 값, 두 번째 for문은 arr2에 열 값, 가장 안쪽에 있는 for문은 arr1과 arr2의 열과 행 값으로 사용된다.

알게된 점

• 이전 문제까지 3차원 배열을 접해본 적이 많지 않아 처음에는 해맸지만 어떤 상황에서 3차원 배열을 써야할 지 약간의 감을 알게된 문제였다.