Lv.2 : 행렬의 곱셈

👉 문제링크

프로그래머스

---

---

🔸 문제 분석 🔸

• 두 2차원 행렬의 곱을 구현하는 문제이다.

🔸 문제 풀이 🔸

• 행렬곱 원리는 다음과 같다.
  • 앞 행렬의 가로 길이(열의 길이)와 뒷 행렬의 세로 길이(행의 길이)가 같아야 곱이 가능하다.
  • 계산 결과 행렬은 앞 행렬의 세로 길이 x 뒷 행렬의 가로 길이 이다.
  • 계산 방법은 다음과 같다.

https://mathbang.net/562#gsc.tab=0

---

🔸 코드 🔸

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

🔸 코드 해석 🔸

• 결과 행렬의 행렬을 row와 col로 저장하고 사용한다.
• row와 col만큼 인덱스를 옮겨가며 결과 행렬을 채워간다.
  • 앞 행렬 arr1은 가로로, 뒷 행렬 arr2는 세로로 움직이며 곱을 계산한다.
  • 앞에서 구한 row, col이 아닌 arr1의 가로 길이(두 행렬이 곱 계산이 가능하려면 같아야 하는 값)만큼 움직인 다는 것에 유의한다.

---

🔸 end 🔸

• 행렬 곱 자체에 대한 지식이 부족해서 수학 이론을 먼저 알아봐야 했다. 간단한 원리인데 고등학교 교과 과정에 빠졌을때 졸업해서 따로 공부하지 않았으니 모르고 있었다.
• 결과 행렬의 값을 계산할때 row혹은 col만큼 반복해서 더하려다 인덱스 벗어남 오류가 났다.
  • 코드 해석 마지막 줄에 있는 것 처럼 앞 행렬의 가로 길이만큼 반복했어야 했다.
• 다른 사람들의 풀이를 보니 sum 변수는 딱히 만들 필요 없이 바로 answer에 넣어도 됐었다.
• 하얀색 배경으로 바꿨는데 눈이 더 편안해져서 당분간 이렇게 쓸 예정이다.