[프로그래머스 Lv.2] 행렬의 곱셈 (Python)

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/12949

프로그래머스

풀이

행렬곱을 구현하면 되는 단순한 문제이다. 행렬곱의 경우 아래와 같은 방식으로 계산한다.

출처: http://www.ktword.co.kr/test/view/view.php?no=5286

이런 계산 방식을 코드로 구현하면 된다.

def solution(arr1, arr2):
    answer = []
    
    for m in range(len(arr1)): # 피연산자(arr1)의 각 행 별로 계산
        arr = []
        for r in range(len(arr2[0])): # 연산자(arr2)의 열의 수만큼 반복
            sum = 0
            for n in range(len(arr2)): # 피연산자(arr1)의 열의 수 == 연산자(arr2)의 행의 수만큼 반복
                sum += arr1[m][n]* arr2[n][r]
            arr.append(sum)
        answer.append(arr)
    
    return answer