[Python] 프로그래머스(Lv2) - 행렬 곱셈

안녕하세요 :)

이 문제는 행렬 곱셈하는 방식 그대로 ... 풀면됩니다

https://programmers.co.kr/learn/courses/30/lessons/12949

여기서 arr1의 행을 r, 행렬 arr1의 열 / arr2의 행을 m, arr2의 열을 n 으로 했습니다.

def solution(arr1, arr2):
    r = len(arr1)
    m = len(arr2)
    n = len(arr2[0])
    answer = [[0] * n for _ in range(r)]
    
    for i in range(r):       
        for j in range(n):
            ab = 0
            for k in range(m):
                ab += arr1[i][k] * arr2[k][j]  
                
            answer[i][j] = ab
        
    return answer