[프로그래머스] Lv.2 행렬의 곱셈 [Python, 파이썬]

• 문제 : https://school.programmers.co.kr/learn/courses/30/lessons/12949
• 행렬 곱셈을 어떻게 하는지 찾아보고 알게 됨, https://mathbang.net/562#gsc.tab=0

def solution(arr1, arr2):
    answer = []
    
    for i in range(len(arr1)):  # arr1의 행 수 n
        tmp = []
        for j in range(len(arr2[0])):   # arr2의 열 수 m
            n = 0
            for k in range(len(arr2)):
                # arr1에서는 오른쪽으로 이동, arr2에서는 아래로 이동하면서 서로 곱함
                n += arr1[i][k] * arr2[k][j]
            tmp.append(n)
        answer.append(tmp)

    # 정답 배열의 크기는 n * m (arr1의 행 수 * arr2의 열 수)
    return answer