[프로그래머스] 행렬의 곱셈

1.  문제

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

프로그래머스

 

2.  접근

1. 행렬의 곱셈은 arr1의 열 * arr2의 행 의 값들을 모아서 더한 값이다.

행렬의 곱셈 예시

-> 결국에는 "시그마" 임을 알아야 한다

2. 그래서 2차원 for loop가 아니라 3차원 for loop가 되어야 문제를 풀 수 있다

 

3.  풀이

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