[프로그래머스 / 파이썬 풀이] 행렬의 곱셈

def solution(arr1, arr2):
    answer = []
    for i in range(len(arr1)):
        num_sum_lst = []
        for j in range(len(arr2[0])):
            num_lst = []
            for k in range(len(arr1[i])):
                num_lst.append(arr1[i][k] * arr2[k][j])
            num_sum_lst.append(sum(num_lst))
        answer.append(num_sum_lst)
        
    return answer

A*B와 C*D의 행렬일 경우 B와 C가 일치해야 행렬의 곱이 이루어진다.

 

이 점을 활용해서 풀이