[프로그래머스/JAVA] 코딩테스트 연습 > 연습문제 > 행렬의 곱셈 (by 다알쥐AI)

틱톡라이트 초대링크 https://lite.tiktok.com/t/ZSjqX7SfC/

[기간 한정] 보는 재미로, 버는 재미로! 영상만 봐도, 좋아요만 눌러도, 검색만 해도, 포인트가 차

[행렬의 곱셈] 코딩테스트 연습 > 연습문제 > 행렬의 곱셈

 
 
※ 주의 : 문제풀이 방법은 다양합니다. 참고만 해주세요 ※
[문제설명]

2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 함수, solution을 완성해주세요.

제한 조건

• 행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.
• 행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.
• 곱할 수 있는 배열만 주어집니다.

입출력 예

---

 

 

 

[간단설명]
행렬의 곱셈(행렬 arr1 x 행렬 arr2) 구하기
 
[접근방법]
1. arr1의 열과 arr2의 행을 곱한다.
2. arr1의 열 길이만큼 반복해서 더한다.
3. answer 행렬에 arr1 * arr2 값을 넣어준다.
4. arr2의 열 길이만큼 반복한다.
5. arr1의 행 길이만큼 반복한다.
6. answer(곱셈 행렬) 리턴
 
[주의사항]
- answer(곱셈 행렬) 의 크기 지정을 올바르게 한다.
- 반복문의 행과 열의 길이를 올바르게 한다.
ㄴ 런타임 오류 주의
 
[소스공개]

class Solution {
    public int[][] solution(int[][] arr1, int[][] arr2) {
/*
        행렬이란게, arr1[열] x arr2[행]을 곱해서 더한 값이다.
        즉, 1번방향 x 2번방향
        
 1번방향 ----> 2번방향
        {1,4}   |         {1x3+4x3, 1x3+4x3}    {15,15}
        {3,2} x | {3,3} = {3x3+2x3, 3x3+2x3} => {15,15}
        {4,1}   V {3,3}   {4x3+1x3, 4x3+1x3}    {15,15}
*/
        // arr1의 행, arr2의 열
        int[][] answer = new int[arr1.length][arr2[0].length];
        
        for(int k=0;k<arr1.length;k++){ //arr1의 행
            for(int i=0;i<arr2[0].length;i++){ // arr2의 열
                int sum = 0;
                for(int j=0;j<arr1[0].length;j++){ // arr1의 열
                    sum += arr1[k][j]*arr2[j][i];
                }
                answer[k][i] = sum; // arr1의 행, arr2의 열
            }
        }
        // 행렬의 곱셈 리턴
        return answer;
    }
}

 
[실행결과]