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

문제 링크 : https://programmers.co.kr/learn/courses/30/lessons/12949
레벨 : Level 2
분류 : 연습문제

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

이 문제는 배열의 움직임을 잘 컨트롤해야 하는 문제이다.

손으로 직접 해보기 전까지 이해하기 쉽지 않았지만 하나씩 차근차근하다 보면 문제를 해결할 수 있다.

다차원 배열의 곱셈은 A배열의 행과 B배열의 열의 개수만큼 결과가 나온다.

예를 들면 A배열이 3*2이고 B배열이 2*2일 때 결과 배열은 3*2이다.

 

3중 배열의 i, j, k의 움직임을 자세히 봐야한다.

 

아래와 같이 소스를 공유합니다.

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