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

문제 설명

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


제한 조건

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

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

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


입출력 예



문제풀이

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

행렬의 곱셈 - 위키백과

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


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

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

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

아래 주석을 참고하세요.

  1. import java.util.*;
  2. class Solution {
  3. public int[][] solution(int[][] arr1, int[][] arr2) {
  4. int[][] answer = {};
  5. answer = new int[arr1.length][arr2[0].length];
  6. for ( int i=0; i<arr1.length; i++ ){
  7. for( int k=0; k<arr2[0].length; k++ ){
  8. int sum = 0;
  9. // arr1.length, arr1[0].length -> 실행 결과는 같으나,
  10. // 배열의 길이에 따라 실행시간이 O(N^2)이 되기 때문에 ex) arr1.length * arr1.length
  11. // arr1[i]의 length로 실행.
  12. for ( int j=0; j<arr1[i].length; j++ ){
  13. sum += arr1[i][j] * arr2[j][k];
  14. }
  15. // System.out.println( sum );
  16. // System.out.println ( i + " " + k);
  17. answer[i][k] = sum;
  18. }
  19. }
  20. return answer;
  21. }
  22. }



* 파트너스 활동을 통해 일정액의 수수료를 제공받을 수 있음
작성자 소개
초이 프로필
WrapUp 블로거

초이

반려견을 좋아하고, 차를 좋아하고, 여행을 좋아하고, 맛집을 찾아 즐기는 웹 개발자 입니다^^