[java] level2 / 연습문제 / 행렬의 곱셈2

[문제 설명]

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]]

---

[풀이]

행렬의 곱 : 

[a1,b1]                              [a1*c1 + b1*c2,  a1*d1+b1*d2]

[a2,b2]    x    [c1,d1]     =  [a2*c1 + b2*c2,  a2*d1+b2*d2]

[a3,b3]         [c2,d2]         [a3*c1 + b3*c2,  a3*d1+b3*d2]

 

class Solution {
    public 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 < arr1[0].length; j++) {
				for (int k = 0; k < arr2[0].length; k++) {
					answer[i][k] += arr1[i][j] * arr2[j][k];
				}
			}
		}
        return answer;
    }
}

** 삼중 for문 사용

1) answer 배열은 [arr1.length][arr2[0].length]로 선언

 

2) arr1.length 만큼 for문 반복

3) arr1[0].length 만큼 for문 반복

4) arr2[0].length 만큼 for문 반복

 

5) answer 배열에 행렬의 곱 공식을 대입하여

곱셈하여 더해줌.

 

6) answer 반환.