프로그래머스/연습문제 Level2
프로그래머스 연습문제 LV2 - 행렬의 곱셈
맴썰
2021. 9. 14. 18:52
문제 설명
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]] |
class Solution {
public int[][] solution(int[][] arr1, int[][] arr2) {
int xlen = arr1.length; int mlen = arr2.length; int ylen = arr2[0].length;
int[][] answer = new int[xlen][ylen];
int sum = 0; int idx = 0; int aidx= 0;
for(int i=0; i<xlen; i++){
for(int j=0; j<ylen;j++){
sum = 0 ;
for(int k=0; k<mlen; k++){
sum += arr1[i][k]*arr2[k][j];
}
answer[i][j] = sum;
}
}
return answer;
}
}