행렬의 곱셈 - level2 (python,java)

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12949?language=java

프로그래머스

풀이

행렬의 곱셈 방식을 알고 있다면 그 식을 3중 for문을 이용해서 구현할 수 있다.

 

 

코드

python

def solution(arr1, arr2):
    answer = [[0]*len(arr2[0]) for _ in range(len(arr1))]


    for i in range(len(arr1)):
        for j in range(len(arr2[0])):
            for k in range(len(arr2)):
                answer[i][j]+=arr1[i][k]*arr2[k][j]

    return answer

 

java

class Solution {
    public int[][] solution(int[][] arr1, int[][] arr2) {
        int a = arr1.length;
        int b= arr2[0].length;
        int[][] answer = new int[a][b];
        
        
        for(int i=0; i<arr1.length; i++){
            for(int j=0; j<arr2[0].length; j++){
                for(int k=0; k<arr2.length; k++){
                    answer[i][j] +=arr1[i][k]*arr2[k][j];
                }
            }
        }
        return answer;
    }
}