JAVA 알고리즘 - 행렬의 곱셈 | 어렵다 어려워

문제 설명

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

제한 조건

행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.

행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.

곱할 수 있는 배열만 주어집니다.

입출력 예

문제풀이

이 문제를 풀기 전에 행렬의 곱셈을 이해해야 합니다.

행렬의 곱셈 - 위키백과

위키백과의 행렬 곱셈의 표기와 정의, 행렬곱의 성분을 이해하고 시작하시길 바랍니다.

문제를 풀다가, 열의 길이 반복문을 돌릴 때 실행결과는 통과했는데 채점 때 실행시간 초과, 런타임 오류가 발생했네요.

로직에는 이상이 없는 것 같은데...... 다른 방법이 있나 몇 시간을 고생한 것 같네요.

결론은 열의 길이를 행의 i 값으로 length 를 지정해주면 된다는거. 

아래 주석을 참고하세요.

import java.util.*;
class Solution {
    public int[][] solution(int[][] arr1, int[][] arr2) {
        int[][] answer = {};
        answer = new int[arr1.length][arr2[0].length];
        for ( int i=0; i<arr1.length; i++ ){
            for( int k=0; k<arr2[0].length; k++ ){
                int sum = 0;
                
                // arr1.length, arr1[0].length -> 실행 결과는 같으나,
                // 배열의 길이에 따라 실행시간이 O(N^2)이 되기 때문에 ex) arr1.length * arr1.length
                // arr1[i]의 length로 실행.
                for ( int j=0; j<arr1[i].length; j++ ){
                    sum += arr1[i][j] * arr2[j][k];
                }
                // System.out.println( sum );
                // System.out.println ( i + " " + k);
                answer[i][k] = sum;
            }
        }
            
        return answer;
    }
    
}