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

[문제 설명]

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

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

 

[풀이 과정]

행렬A와 행렬B 를 곱하기 위해서는 두 행렬이 다음 성질을 만족해야 한다. 

출처 : https://mathbang.net/562

1. 행렬A의 행의 길이(k)와 행렬B의 열의 길이(k)가 같아야 한다. 

2. 두 행렬의 곱한 결과 행렬은 가로 세로가 (행렬B의 행 * 행렬A의 열) 이다. (m * n)

3. A * B 가 만족하듯이 B * A가 만족하려면 그림처럼 두 행렬의 k 부분이 같아야 한다. ( 다르면 곱할 수 없음)

 

코드로 구현할 때는 이 유튜브 알고리즘을 보고 도움을 받았다. 

https://www.youtube.com/watch?v=yCWLN4tSxjk   

 

[JAVA code]

package com.algorithm.level2;

import java.util.Arrays;

public class Test20 {

    public static void main(String[] args) {
        int[][] arr1 = {{1, 4}, {3, 2}, {4, 1}};
        int[][] arr2 = {{3, 3}, {3, 3}};

        int[][] answer = solution(arr1, arr2);
        for (int i = 0; i < answer.length; i++) {
            System.out.println(Arrays.toString(answer[i]));
        }
    }

    public static int[][] solution(int[][] arr1, int[][] arr2) {
        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;
    }
}